diff --git a/example/yolov3_coco2017/README.md b/model_zoo/yolov3/README.md similarity index 100% rename from example/yolov3_coco2017/README.md rename to model_zoo/yolov3/README.md diff --git a/example/yolov3_coco2017/eval.py b/model_zoo/yolov3/eval.py similarity index 95% rename from example/yolov3_coco2017/eval.py rename to model_zoo/yolov3/eval.py index 6e6d358248..433ae834ba 100644 --- a/example/yolov3_coco2017/eval.py +++ b/model_zoo/yolov3/eval.py @@ -19,10 +19,10 @@ import argparse import time from mindspore import context, Tensor from mindspore.train.serialization import load_checkpoint, load_param_into_net -from mindspore.model_zoo.yolov3 import yolov3_resnet18, YoloWithEval -from dataset import create_yolo_dataset, data_to_mindrecord_byte_image -from config import ConfigYOLOV3ResNet18 -from util import metrics +from src.yolov3 import yolov3_resnet18, YoloWithEval +from src.dataset import create_yolo_dataset, data_to_mindrecord_byte_image +from src.config import ConfigYOLOV3ResNet18 +from src.utils import metrics def yolo_eval(dataset_path, ckpt_path): """Yolov3 evaluation.""" diff --git a/example/yolov3_coco2017/run_distribute_train.sh b/model_zoo/yolov3/scripts/run_distribute_train.sh similarity index 94% rename from example/yolov3_coco2017/run_distribute_train.sh rename to model_zoo/yolov3/scripts/run_distribute_train.sh index 0b764419d2..eeda5077e9 100644 --- a/example/yolov3_coco2017/run_distribute_train.sh +++ b/model_zoo/yolov3/scripts/run_distribute_train.sh @@ -45,6 +45,9 @@ echo "After running the scipt, the network runs in the background. The log will export MINDSPORE_HCCL_CONFIG_PATH=$6 export RANK_SIZE=$1 +BASE_PATH=$(cd "`dirname $0`" || exit; pwd) +cd $BASE_PATH/../ || exit + for((i=0;i