提交 37a81c4f 编写于 作者: C chengxianbin

clear pylint for yolov3

上级 14241786
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册