diff --git a/model_zoo/ssd/README.md b/model_zoo/official/cv/ssd/README.md similarity index 100% rename from model_zoo/ssd/README.md rename to model_zoo/official/cv/ssd/README.md diff --git a/model_zoo/ssd/eval.py b/model_zoo/official/cv/ssd/eval.py similarity index 100% rename from model_zoo/ssd/eval.py rename to model_zoo/official/cv/ssd/eval.py diff --git a/model_zoo/ssd/scripts/run_distribute_train.sh b/model_zoo/official/cv/ssd/scripts/run_distribute_train.sh similarity index 100% rename from model_zoo/ssd/scripts/run_distribute_train.sh rename to model_zoo/official/cv/ssd/scripts/run_distribute_train.sh diff --git a/model_zoo/ssd/src/__init__.py b/model_zoo/official/cv/ssd/src/__init__.py similarity index 100% rename from model_zoo/ssd/src/__init__.py rename to model_zoo/official/cv/ssd/src/__init__.py diff --git a/model_zoo/ssd/src/box_utils.py b/model_zoo/official/cv/ssd/src/box_utils.py similarity index 100% rename from model_zoo/ssd/src/box_utils.py rename to model_zoo/official/cv/ssd/src/box_utils.py diff --git a/model_zoo/ssd/src/coco_eval.py b/model_zoo/official/cv/ssd/src/coco_eval.py similarity index 100% rename from model_zoo/ssd/src/coco_eval.py rename to model_zoo/official/cv/ssd/src/coco_eval.py diff --git a/model_zoo/ssd/src/config.py b/model_zoo/official/cv/ssd/src/config.py similarity index 100% rename from model_zoo/ssd/src/config.py rename to model_zoo/official/cv/ssd/src/config.py diff --git a/model_zoo/ssd/src/dataset.py b/model_zoo/official/cv/ssd/src/dataset.py similarity index 100% rename from model_zoo/ssd/src/dataset.py rename to model_zoo/official/cv/ssd/src/dataset.py diff --git a/model_zoo/ssd/src/init_params.py b/model_zoo/official/cv/ssd/src/init_params.py similarity index 100% rename from model_zoo/ssd/src/init_params.py rename to model_zoo/official/cv/ssd/src/init_params.py diff --git a/model_zoo/ssd/src/lr_schedule.py b/model_zoo/official/cv/ssd/src/lr_schedule.py similarity index 100% rename from model_zoo/ssd/src/lr_schedule.py rename to model_zoo/official/cv/ssd/src/lr_schedule.py diff --git a/model_zoo/ssd/src/ssd.py b/model_zoo/official/cv/ssd/src/ssd.py similarity index 100% rename from model_zoo/ssd/src/ssd.py rename to model_zoo/official/cv/ssd/src/ssd.py diff --git a/model_zoo/ssd/train.py b/model_zoo/official/cv/ssd/train.py similarity index 100% rename from model_zoo/ssd/train.py rename to model_zoo/official/cv/ssd/train.py diff --git a/model_zoo/yolov3_resnet18/README.md b/model_zoo/official/cv/yolov3_resnet18/README.md similarity index 100% rename from model_zoo/yolov3_resnet18/README.md rename to model_zoo/official/cv/yolov3_resnet18/README.md diff --git a/model_zoo/yolov3_resnet18/eval.py b/model_zoo/official/cv/yolov3_resnet18/eval.py similarity index 100% rename from model_zoo/yolov3_resnet18/eval.py rename to model_zoo/official/cv/yolov3_resnet18/eval.py diff --git a/model_zoo/yolov3_resnet18/scripts/run_distribute_train.sh b/model_zoo/official/cv/yolov3_resnet18/scripts/run_distribute_train.sh similarity index 100% rename from model_zoo/yolov3_resnet18/scripts/run_distribute_train.sh rename to model_zoo/official/cv/yolov3_resnet18/scripts/run_distribute_train.sh diff --git a/model_zoo/yolov3_resnet18/scripts/run_eval.sh b/model_zoo/official/cv/yolov3_resnet18/scripts/run_eval.sh similarity index 100% rename from model_zoo/yolov3_resnet18/scripts/run_eval.sh rename to model_zoo/official/cv/yolov3_resnet18/scripts/run_eval.sh diff --git a/model_zoo/yolov3_resnet18/scripts/run_standalone_train.sh b/model_zoo/official/cv/yolov3_resnet18/scripts/run_standalone_train.sh similarity index 100% rename from model_zoo/yolov3_resnet18/scripts/run_standalone_train.sh rename to model_zoo/official/cv/yolov3_resnet18/scripts/run_standalone_train.sh diff --git a/model_zoo/yolov3_resnet18/src/config.py b/model_zoo/official/cv/yolov3_resnet18/src/config.py similarity index 100% rename from model_zoo/yolov3_resnet18/src/config.py rename to model_zoo/official/cv/yolov3_resnet18/src/config.py diff --git a/model_zoo/yolov3_resnet18/src/dataset.py b/model_zoo/official/cv/yolov3_resnet18/src/dataset.py similarity index 100% rename from model_zoo/yolov3_resnet18/src/dataset.py rename to model_zoo/official/cv/yolov3_resnet18/src/dataset.py diff --git a/model_zoo/yolov3_resnet18/src/utils.py b/model_zoo/official/cv/yolov3_resnet18/src/utils.py similarity index 100% rename from model_zoo/yolov3_resnet18/src/utils.py rename to model_zoo/official/cv/yolov3_resnet18/src/utils.py diff --git a/model_zoo/yolov3_resnet18/src/yolov3.py b/model_zoo/official/cv/yolov3_resnet18/src/yolov3.py similarity index 100% rename from model_zoo/yolov3_resnet18/src/yolov3.py rename to model_zoo/official/cv/yolov3_resnet18/src/yolov3.py diff --git a/model_zoo/yolov3_resnet18/train.py b/model_zoo/official/cv/yolov3_resnet18/train.py similarity index 98% rename from model_zoo/yolov3_resnet18/train.py rename to model_zoo/official/cv/yolov3_resnet18/train.py index 6a139627cf4f3879f8864da1ddcc19ee2e4929fd..761a94c454a151a8b523f60475ab1b8182cd62a3 100644 --- a/model_zoo/yolov3_resnet18/train.py +++ b/model_zoo/official/cv/yolov3_resnet18/train.py @@ -111,8 +111,8 @@ def main(): data_to_mindrecord_byte_image(args_opt.image_dir, args_opt.anno_path, args_opt.mindrecord_dir, - prefix=prefix, - file_num=8) + prefix, + 8) print("Create Mindrecord Done, at {}".format(args_opt.mindrecord_dir)) else: print("image_dir or anno_path not exits.")