From 37a81c4f7efea4dcfc3133d13136d5307862ec49 Mon Sep 17 00:00:00 2001 From: chengxianbin Date: Fri, 19 Jun 2020 10:28:07 +0800 Subject: [PATCH] clear pylint for yolov3 --- model_zoo/yolov3/src/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/yolov3/src/dataset.py b/model_zoo/yolov3/src/dataset.py index 22e3a1dad2..f85b442209 100644 --- 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)