change directory of YOLOv3-ResNet18 and SSD from model_zoo to model_zoo/official/cv
This commit is contained in:
parent
28c8a5cc26
commit
8fc545d40b
|
@ -111,8 +111,8 @@ def main():
|
||||||
data_to_mindrecord_byte_image(args_opt.image_dir,
|
data_to_mindrecord_byte_image(args_opt.image_dir,
|
||||||
args_opt.anno_path,
|
args_opt.anno_path,
|
||||||
args_opt.mindrecord_dir,
|
args_opt.mindrecord_dir,
|
||||||
prefix=prefix,
|
prefix,
|
||||||
file_num=8)
|
8)
|
||||||
print("Create Mindrecord Done, at {}".format(args_opt.mindrecord_dir))
|
print("Create Mindrecord Done, at {}".format(args_opt.mindrecord_dir))
|
||||||
else:
|
else:
|
||||||
print("image_dir or anno_path not exits.")
|
print("image_dir or anno_path not exits.")
|
Loading…
Reference in New Issue