From 9a2f88bedc9c68d4f8ba7ac7042bdade437af7be Mon Sep 17 00:00:00 2001 From: Dang Qingqing Date: Wed, 11 Apr 2018 22:03:01 +0800 Subject: [PATCH] Change the pre-trained model URL. --- fluid/object_detection/.gitignore | 8 ++++++++ fluid/object_detection/pretrained/download_coco.sh | 4 ++-- fluid/object_detection/pretrained/download_imagenet.sh | 4 ++-- fluid/object_detection/train.py | 2 +- 4 files changed, 13 insertions(+), 5 deletions(-) create mode 100644 fluid/object_detection/.gitignore diff --git a/fluid/object_detection/.gitignore b/fluid/object_detection/.gitignore new file mode 100644 index 00000000..3321aa10 --- /dev/null +++ b/fluid/object_detection/.gitignore @@ -0,0 +1,8 @@ +./data/pascalvoc/VOCdevkit/ +data/pascalvoc/test.txt +data/pascalvoc/trainval.txt +pretrained/ssd_mobilenet_v1_coco.tar.gz +pretrained/ssd_mobilenet_v1_coco +pretrained/mobilenet_v1_imagenet.tar.gz +pretrained/mobilenet_v1_imagenet +log* diff --git a/fluid/object_detection/pretrained/download_coco.sh b/fluid/object_detection/pretrained/download_coco.sh index 41684b11..1cd1836f 100644 --- a/fluid/object_detection/pretrained/download_coco.sh +++ b/fluid/object_detection/pretrained/download_coco.sh @@ -3,6 +3,6 @@ cd "$DIR" # Download the data. echo "Downloading..." -wget http://paddlemodels.bj.bcebos.com/ssd_mobilenet_coco.tar.gz +wget http://paddlemodels.bj.bcebos.com/ssd_mobilenet_v1_coco.tar.gz echo "Extractint..." -tar -xf ssd_mobilenet_coco.tar.gz +tar -xf ssd_mobilenet_v1_coco.tar.gz diff --git a/fluid/object_detection/pretrained/download_imagenet.sh b/fluid/object_detection/pretrained/download_imagenet.sh index 597cc8a4..eb7c6767 100644 --- a/fluid/object_detection/pretrained/download_imagenet.sh +++ b/fluid/object_detection/pretrained/download_imagenet.sh @@ -3,6 +3,6 @@ cd "$DIR" # Download the data. echo "Downloading..." -wget http://paddlemodels.bj.bcebos.com/mobilenet_imagenet.tar.gz +wget http://paddlemodels.bj.bcebos.com/mobilenet_v1_imagenet.tar.gz echo "Extractint..." -tar -xf ssd_mobilenet_imagenet.tar.gz +tar -xf mobilenet_v1_imagenet.tar.gz diff --git a/fluid/object_detection/train.py b/fluid/object_detection/train.py index 9f89a832..8610500e 100644 --- a/fluid/object_detection/train.py +++ b/fluid/object_detection/train.py @@ -20,7 +20,7 @@ add_arg('parallel', bool, True, "Whether use parallel training.") add_arg('use_gpu', bool, True, "Whether use GPU.") add_arg('dataset', str, 'pascalvoc', "coco or pascalvoc.") add_arg('model_save_dir', str, 'model', "The path to save model.") -add_arg('pretrained_model', str, 'pretrained/ssd_mobilenet_coco/', "The init model path.") +add_arg('pretrained_model', str, 'pretrained/ssd_mobilenet_v1_coco/', "The init model path.") add_arg('apply_distort', bool, True, "Whether apply distort") add_arg('apply_expand', bool, False, "Whether appley expand") add_arg('resize_h', int, 300, "resize image size") -- GitLab