Browse Source

!2335 clear pylint warning for yolov3

Merge pull request !2335 from chengxb7532/cxb_st
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
b2ac1e3dbe
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/yolov3/src/dataset.py

+ 1
- 1
model_zoo/yolov3/src/dataset.py View File

@@ -268,7 +268,7 @@ def filter_valid_data(image_dir, anno_path):
return image_files, image_anno_dict 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.""" """Create MindRecord file by image_dir and anno_path."""
mindrecord_path = os.path.join(mindrecord_dir, prefix) mindrecord_path = os.path.join(mindrecord_dir, prefix)
writer = FileWriter(mindrecord_path, file_num) writer = FileWriter(mindrecord_path, file_num)


Loading…
Cancel
Save