diff --git a/fluid/object_detection/.gitignore b/fluid/object_detection/.gitignore new file mode 100644 index 0000000000000000000000000000000000000000..3321aa105e8c63b5ba915782fd69bc90debbf56c --- /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 41684b116e327a2e052414f0b933f79b61acc1a2..1cd1836f7c6e32f9f308a0c9a29d10efbc6f183f 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 597cc8a44c55b29f63f3fc374dc252b515f310c9..eb7c6767d9f9585342c2ba89a2f28f070d1351c2 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 9f89a832b17a9b2b6bec193cf5cacddcd95e1a53..8610500edf051fb8cfd195add99402e8bd54590f 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")