├── README.md ├── data ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-36.pyc │ ├── coco.cpython-36.pyc │ ├── config.cpython-36.pyc │ └── voc0712.cpython-36.pyc ├── coco.py ├── coco │ └── coco_labels.txt ├── config.py ├── example.jpg ├── scripts │ ├── COCO2014.sh │ ├── VOC2007.sh │ └── VOC2012.sh └── voc0712.py ├── detection.py ├── eval.py ├── focal_loss.py ├── layers ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-36.pyc │ └── box_utils.cpython-36.pyc ├── box_utils.py ├── functions │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── detection.cpython-36.pyc │ │ └── prior_box.cpython-36.pyc │ ├── detection.py │ └── prior_box.py └── modules │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-36.pyc │ ├── l2norm.cpython-36.pyc │ └── multibox_loss.cpython-36.pyc │ ├── l2norm.py │ └── multibox_loss.py ├── mobilenetv2.py ├── out_images └── dog.jpg ├── output_videos └── out_video ├── result └── result ├── ssd_mobilenetv2.py ├── test_images └── dog.jpg ├── test_videos └── videos ├── train.py ├── utils ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-36.pyc │ └── augmentations.cpython-36.pyc └── augmentations.py └── weights ├── pre_trained model put here └── ssd_mobilenetv2_300 └── trained model /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /data/__pycache__/coco.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/__pycache__/coco.cpython-36.pyc -------------------------------------------------------------------------------- /data/__pycache__/config.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/__pycache__/config.cpython-36.pyc -------------------------------------------------------------------------------- /data/__pycache__/voc0712.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/__pycache__/voc0712.cpython-36.pyc -------------------------------------------------------------------------------- /data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/coco.py -------------------------------------------------------------------------------- /data/coco/coco_labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/coco/coco_labels.txt -------------------------------------------------------------------------------- /data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/config.py -------------------------------------------------------------------------------- /data/example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/example.jpg -------------------------------------------------------------------------------- /data/scripts/COCO2014.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/scripts/COCO2014.sh -------------------------------------------------------------------------------- /data/scripts/VOC2007.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/scripts/VOC2007.sh -------------------------------------------------------------------------------- /data/scripts/VOC2012.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/scripts/VOC2012.sh -------------------------------------------------------------------------------- /data/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/data/voc0712.py -------------------------------------------------------------------------------- /detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/detection.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/eval.py -------------------------------------------------------------------------------- /focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/focal_loss.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/__init__.py -------------------------------------------------------------------------------- /layers/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /layers/__pycache__/box_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/__pycache__/box_utils.cpython-36.pyc -------------------------------------------------------------------------------- /layers/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/box_utils.py -------------------------------------------------------------------------------- /layers/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/functions/__init__.py -------------------------------------------------------------------------------- /layers/functions/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/functions/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /layers/functions/__pycache__/detection.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/functions/__pycache__/detection.cpython-36.pyc -------------------------------------------------------------------------------- /layers/functions/__pycache__/prior_box.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/functions/__pycache__/prior_box.cpython-36.pyc -------------------------------------------------------------------------------- /layers/functions/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/functions/detection.py -------------------------------------------------------------------------------- /layers/functions/prior_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/functions/prior_box.py -------------------------------------------------------------------------------- /layers/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/modules/__init__.py -------------------------------------------------------------------------------- /layers/modules/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/modules/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /layers/modules/__pycache__/l2norm.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/modules/__pycache__/l2norm.cpython-36.pyc -------------------------------------------------------------------------------- /layers/modules/__pycache__/multibox_loss.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/modules/__pycache__/multibox_loss.cpython-36.pyc -------------------------------------------------------------------------------- /layers/modules/l2norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/modules/l2norm.py -------------------------------------------------------------------------------- /layers/modules/multibox_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/layers/modules/multibox_loss.py -------------------------------------------------------------------------------- /mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/mobilenetv2.py -------------------------------------------------------------------------------- /out_images/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/out_images/dog.jpg -------------------------------------------------------------------------------- /output_videos/out_video: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /result/result: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /ssd_mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/ssd_mobilenetv2.py -------------------------------------------------------------------------------- /test_images/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/test_images/dog.jpg -------------------------------------------------------------------------------- /test_videos/videos: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/utils/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/augmentations.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/utils/__pycache__/augmentations.cpython-36.pyc -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /weights/pre_trained model put here: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /weights/ssd_mobilenetv2_300/trained model: -------------------------------------------------------------------------------- 1 | 2 | --------------------------------------------------------------------------------