提交 b2ac1e3d 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2335 clear pylint warning for yolov3

Merge pull request !2335 from chengxb7532/cxb_st
......@@ -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.
先完成此消息的编辑!
想要评论请 注册