├── .gitattributes ├── .gitignore ├── LICENSE ├── README.md ├── coco_voc.txt ├── data ├── .config.py.swp ├── __init__.py ├── coco.py ├── config.py ├── data_augment.py ├── scripts │ ├── VOC2007.sh │ └── VOC2012.sh ├── voc0712.py └── voc_eval.py ├── demo └── live.py ├── examples ├── result1.jpg └── result4.jpg ├── layers ├── __init__.py ├── functions │ ├── __init__.py │ ├── detection.py │ └── prior_box.py └── modules │ ├── __init__.py │ ├── l2norm.py │ ├── multibox_loss.py │ └── refine_multibox_loss.py ├── models ├── FRFBSSD_vgg.py ├── FSSD_mobile.py ├── FSSD_vgg.py ├── MOD_Resnet34.py ├── MOD_vgg_1109.py ├── MOD_vgg_1109_2.py ├── MOD_vgg_1111.py ├── MOD_vgg_1125.py ├── MOD_vgg_1125_2.py ├── MOD_vgg_1129.py ├── RFB_Net_E_vgg.py ├── RFB_Net_mobile.py ├── RFB_Net_vgg.py ├── RefineDet_vgg.py ├── RefineSSD_vgg.py ├── SSD_vgg.py ├── __init__.py ├── base_models.py └── mobilenet.py ├── pcb_test.py ├── refinedet_train_test.py ├── train_test.py ├── utils ├── nms │ ├── gpu_nms.cpp │ └── gpu_nms.hpp └── nms_wrapper.py └── weights └── vgg16_reducedfc.pth /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/README.md -------------------------------------------------------------------------------- /coco_voc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/coco_voc.txt -------------------------------------------------------------------------------- /data/.config.py.swp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/data/.config.py.swp -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/data/coco.py -------------------------------------------------------------------------------- /data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/data/config.py -------------------------------------------------------------------------------- /data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/data/data_augment.py -------------------------------------------------------------------------------- /data/scripts/VOC2007.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/data/scripts/VOC2007.sh -------------------------------------------------------------------------------- /data/scripts/VOC2012.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/data/scripts/VOC2012.sh -------------------------------------------------------------------------------- /data/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/data/voc0712.py -------------------------------------------------------------------------------- /data/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/data/voc_eval.py -------------------------------------------------------------------------------- /demo/live.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/demo/live.py -------------------------------------------------------------------------------- /examples/result1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/examples/result1.jpg -------------------------------------------------------------------------------- /examples/result4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/examples/result4.jpg -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/layers/__init__.py -------------------------------------------------------------------------------- /layers/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/layers/functions/__init__.py -------------------------------------------------------------------------------- /layers/functions/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/layers/functions/detection.py -------------------------------------------------------------------------------- /layers/functions/prior_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/layers/functions/prior_box.py -------------------------------------------------------------------------------- /layers/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/layers/modules/__init__.py -------------------------------------------------------------------------------- /layers/modules/l2norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/layers/modules/l2norm.py -------------------------------------------------------------------------------- /layers/modules/multibox_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/layers/modules/multibox_loss.py -------------------------------------------------------------------------------- /layers/modules/refine_multibox_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/layers/modules/refine_multibox_loss.py -------------------------------------------------------------------------------- /models/FRFBSSD_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/FRFBSSD_vgg.py -------------------------------------------------------------------------------- /models/FSSD_mobile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/FSSD_mobile.py -------------------------------------------------------------------------------- /models/FSSD_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/FSSD_vgg.py -------------------------------------------------------------------------------- /models/MOD_Resnet34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/MOD_Resnet34.py -------------------------------------------------------------------------------- /models/MOD_vgg_1109.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/MOD_vgg_1109.py -------------------------------------------------------------------------------- /models/MOD_vgg_1109_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/MOD_vgg_1109_2.py -------------------------------------------------------------------------------- /models/MOD_vgg_1111.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/MOD_vgg_1111.py -------------------------------------------------------------------------------- /models/MOD_vgg_1125.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/MOD_vgg_1125.py -------------------------------------------------------------------------------- /models/MOD_vgg_1125_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/MOD_vgg_1125_2.py -------------------------------------------------------------------------------- /models/MOD_vgg_1129.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/MOD_vgg_1129.py -------------------------------------------------------------------------------- /models/RFB_Net_E_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/RFB_Net_E_vgg.py -------------------------------------------------------------------------------- /models/RFB_Net_mobile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/RFB_Net_mobile.py -------------------------------------------------------------------------------- /models/RFB_Net_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/RFB_Net_vgg.py -------------------------------------------------------------------------------- /models/RefineDet_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/RefineDet_vgg.py -------------------------------------------------------------------------------- /models/RefineSSD_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/RefineSSD_vgg.py -------------------------------------------------------------------------------- /models/SSD_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/SSD_vgg.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/base_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/base_models.py -------------------------------------------------------------------------------- /models/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/models/mobilenet.py -------------------------------------------------------------------------------- /pcb_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/pcb_test.py -------------------------------------------------------------------------------- /refinedet_train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/refinedet_train_test.py -------------------------------------------------------------------------------- /train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/train_test.py -------------------------------------------------------------------------------- /utils/nms/gpu_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/utils/nms/gpu_nms.cpp -------------------------------------------------------------------------------- /utils/nms/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/utils/nms/gpu_nms.hpp -------------------------------------------------------------------------------- /utils/nms_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/utils/nms_wrapper.py -------------------------------------------------------------------------------- /weights/vgg16_reducedfc.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luissen/SSDT-A-single-shot-detector-for-PCB--defects/HEAD/weights/vgg16_reducedfc.pth --------------------------------------------------------------------------------