clear pylint for yolov3

5 years ago · 37a81c4f7e
parent 1424178601
commit 37a81c4f7e
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/yolov3/src/dataset.py
+++ b/model_zoo/yolov3/src/dataset.py
@ -268,7 +268,7 @@ def filter_valid_data(image_dir, anno_path):
    return image_files, image_anno_dict


-def data_to_mindrecord_byte_image(image_dir, anno_path, mindrecord_dir, prefix="yolo.mindrecord", file_num=8):
+def data_to_mindrecord_byte_image(image_dir, anno_path, mindrecord_dir, prefix, file_num):
    """Create MindRecord file by image_dir and anno_path."""
    mindrecord_path = os.path.join(mindrecord_dir, prefix)
    writer = FileWriter(mindrecord_path, file_num)