From 8fc545d40b05cdaf00b9631fce33b0a5f645dfb6 Mon Sep 17 00:00:00 2001 From: chengxianbin Date: Tue, 21 Jul 2020 20:16:45 +0800 Subject: [PATCH] change directory of YOLOv3-ResNet18 and SSD from model_zoo to model_zoo/official/cv --- model_zoo/{ => official/cv}/ssd/README.md | 0 model_zoo/{ => official/cv}/ssd/eval.py | 0 .../{ => official/cv}/ssd/scripts/run_distribute_train.sh | 0 model_zoo/{ => official/cv}/ssd/src/__init__.py | 0 model_zoo/{ => official/cv}/ssd/src/box_utils.py | 0 model_zoo/{ => official/cv}/ssd/src/coco_eval.py | 0 model_zoo/{ => official/cv}/ssd/src/config.py | 0 model_zoo/{ => official/cv}/ssd/src/dataset.py | 0 model_zoo/{ => official/cv}/ssd/src/init_params.py | 0 model_zoo/{ => official/cv}/ssd/src/lr_schedule.py | 0 model_zoo/{ => official/cv}/ssd/src/ssd.py | 0 model_zoo/{ => official/cv}/ssd/train.py | 0 model_zoo/{ => official/cv}/yolov3_resnet18/README.md | 0 model_zoo/{ => official/cv}/yolov3_resnet18/eval.py | 0 .../cv}/yolov3_resnet18/scripts/run_distribute_train.sh | 0 .../{ => official/cv}/yolov3_resnet18/scripts/run_eval.sh | 0 .../cv}/yolov3_resnet18/scripts/run_standalone_train.sh | 0 model_zoo/{ => official/cv}/yolov3_resnet18/src/config.py | 0 model_zoo/{ => official/cv}/yolov3_resnet18/src/dataset.py | 0 model_zoo/{ => official/cv}/yolov3_resnet18/src/utils.py | 0 model_zoo/{ => official/cv}/yolov3_resnet18/src/yolov3.py | 0 model_zoo/{ => official/cv}/yolov3_resnet18/train.py | 4 ++-- 22 files changed, 2 insertions(+), 2 deletions(-) rename model_zoo/{ => official/cv}/ssd/README.md (100%) rename model_zoo/{ => official/cv}/ssd/eval.py (100%) rename model_zoo/{ => official/cv}/ssd/scripts/run_distribute_train.sh (100%) rename model_zoo/{ => official/cv}/ssd/src/__init__.py (100%) rename model_zoo/{ => official/cv}/ssd/src/box_utils.py (100%) rename model_zoo/{ => official/cv}/ssd/src/coco_eval.py (100%) rename model_zoo/{ => official/cv}/ssd/src/config.py (100%) rename model_zoo/{ => official/cv}/ssd/src/dataset.py (100%) rename model_zoo/{ => official/cv}/ssd/src/init_params.py (100%) rename model_zoo/{ => official/cv}/ssd/src/lr_schedule.py (100%) rename model_zoo/{ => official/cv}/ssd/src/ssd.py (100%) rename model_zoo/{ => official/cv}/ssd/train.py (100%) rename model_zoo/{ => official/cv}/yolov3_resnet18/README.md (100%) rename model_zoo/{ => official/cv}/yolov3_resnet18/eval.py (100%) rename model_zoo/{ => official/cv}/yolov3_resnet18/scripts/run_distribute_train.sh (100%) rename model_zoo/{ => official/cv}/yolov3_resnet18/scripts/run_eval.sh (100%) rename model_zoo/{ => official/cv}/yolov3_resnet18/scripts/run_standalone_train.sh (100%) rename model_zoo/{ => official/cv}/yolov3_resnet18/src/config.py (100%) rename model_zoo/{ => official/cv}/yolov3_resnet18/src/dataset.py (100%) rename model_zoo/{ => official/cv}/yolov3_resnet18/src/utils.py (100%) rename model_zoo/{ => official/cv}/yolov3_resnet18/src/yolov3.py (100%) rename model_zoo/{ => official/cv}/yolov3_resnet18/train.py (98%) 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 6a139627cf..761a94c454 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.")