├── .gitignore ├── LICENSE ├── README.md ├── data └── .gitkeep ├── eval_ssd.py ├── models └── .gitkeep ├── onnx_export.py ├── open_images_classes.txt ├── open_images_downloader.py ├── requirements.txt ├── run_ssd_example.py ├── train_ssd.py └── vision ├── __init__.py ├── datasets ├── __init__.py ├── collation.py ├── generate_vocdata.py ├── open_images.py └── voc_dataset.py ├── nn ├── __init__.py ├── alexnet.py ├── mobilenet.py ├── mobilenet_v2.py ├── multibox_loss.py ├── scaled_l2_norm.py ├── squeezenet.py └── vgg.py ├── prunning ├── __init__.py └── prunner.py ├── ssd ├── __init__.py ├── config │ ├── __init__.py │ ├── mobilenetv1_ssd_config.py │ ├── squeezenet_ssd_config.py │ └── vgg_ssd_config.py ├── data_preprocessing.py ├── fpn_mobilenetv1_ssd.py ├── fpn_ssd.py ├── mobilenet_v2_ssd_lite.py ├── mobilenetv1_ssd.py ├── mobilenetv1_ssd_lite.py ├── predictor.py ├── squeezenet_ssd_lite.py ├── ssd.py └── vgg_ssd.py ├── test ├── __init__.py ├── assets │ └── 000138.jpg └── test_vgg_ssd.py ├── transforms ├── __init__.py └── transforms.py └── utils ├── __init__.py ├── box_utils.py ├── box_utils_numpy.py ├── measurements.py ├── misc.py └── model_book.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/README.md -------------------------------------------------------------------------------- /data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/eval_ssd.py -------------------------------------------------------------------------------- /models/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /onnx_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/onnx_export.py -------------------------------------------------------------------------------- /open_images_classes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/open_images_classes.txt -------------------------------------------------------------------------------- /open_images_downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/open_images_downloader.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | boto3 2 | pandas 3 | urllib3 4 | tensorboard -------------------------------------------------------------------------------- /run_ssd_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/run_ssd_example.py -------------------------------------------------------------------------------- /train_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/train_ssd.py -------------------------------------------------------------------------------- /vision/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vision/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vision/datasets/collation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/datasets/collation.py -------------------------------------------------------------------------------- /vision/datasets/generate_vocdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/datasets/generate_vocdata.py -------------------------------------------------------------------------------- /vision/datasets/open_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/datasets/open_images.py -------------------------------------------------------------------------------- /vision/datasets/voc_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/datasets/voc_dataset.py -------------------------------------------------------------------------------- /vision/nn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vision/nn/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/nn/alexnet.py -------------------------------------------------------------------------------- /vision/nn/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/nn/mobilenet.py -------------------------------------------------------------------------------- /vision/nn/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/nn/mobilenet_v2.py -------------------------------------------------------------------------------- /vision/nn/multibox_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/nn/multibox_loss.py -------------------------------------------------------------------------------- /vision/nn/scaled_l2_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/nn/scaled_l2_norm.py -------------------------------------------------------------------------------- /vision/nn/squeezenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/nn/squeezenet.py -------------------------------------------------------------------------------- /vision/nn/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/nn/vgg.py -------------------------------------------------------------------------------- /vision/prunning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vision/prunning/prunner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/prunning/prunner.py -------------------------------------------------------------------------------- /vision/ssd/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vision/ssd/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vision/ssd/config/mobilenetv1_ssd_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/config/mobilenetv1_ssd_config.py -------------------------------------------------------------------------------- /vision/ssd/config/squeezenet_ssd_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/config/squeezenet_ssd_config.py -------------------------------------------------------------------------------- /vision/ssd/config/vgg_ssd_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/config/vgg_ssd_config.py -------------------------------------------------------------------------------- /vision/ssd/data_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/data_preprocessing.py -------------------------------------------------------------------------------- /vision/ssd/fpn_mobilenetv1_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/fpn_mobilenetv1_ssd.py -------------------------------------------------------------------------------- /vision/ssd/fpn_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/fpn_ssd.py -------------------------------------------------------------------------------- /vision/ssd/mobilenet_v2_ssd_lite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/mobilenet_v2_ssd_lite.py -------------------------------------------------------------------------------- /vision/ssd/mobilenetv1_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/mobilenetv1_ssd.py -------------------------------------------------------------------------------- /vision/ssd/mobilenetv1_ssd_lite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/mobilenetv1_ssd_lite.py -------------------------------------------------------------------------------- /vision/ssd/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/predictor.py -------------------------------------------------------------------------------- /vision/ssd/squeezenet_ssd_lite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/squeezenet_ssd_lite.py -------------------------------------------------------------------------------- /vision/ssd/ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/ssd.py -------------------------------------------------------------------------------- /vision/ssd/vgg_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/ssd/vgg_ssd.py -------------------------------------------------------------------------------- /vision/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vision/test/assets/000138.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/test/assets/000138.jpg -------------------------------------------------------------------------------- /vision/test/test_vgg_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/test/test_vgg_ssd.py -------------------------------------------------------------------------------- /vision/transforms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vision/transforms/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/transforms/transforms.py -------------------------------------------------------------------------------- /vision/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .misc import * 2 | -------------------------------------------------------------------------------- /vision/utils/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/utils/box_utils.py -------------------------------------------------------------------------------- /vision/utils/box_utils_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/utils/box_utils_numpy.py -------------------------------------------------------------------------------- /vision/utils/measurements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/utils/measurements.py -------------------------------------------------------------------------------- /vision/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/utils/misc.py -------------------------------------------------------------------------------- /vision/utils/model_book.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dusty-nv/pytorch-ssd/HEAD/vision/utils/model_book.py --------------------------------------------------------------------------------