├── .gitattributes ├── .github └── stale.yml ├── .gitignore ├── CONTRIBUTING.md ├── ISSUE_TEMPLATE.md ├── LICENSE.txt ├── README.md ├── __init__.py ├── bounding_box_utils ├── __init__.py └── bounding_box_utils.py ├── data_generator ├── __init__.py ├── data_augmentation_chain_constant_input_size.py ├── data_augmentation_chain_original_ssd.py ├── data_augmentation_chain_satellite.py ├── data_augmentation_chain_variable_input_size.py ├── object_detection_2d_data_generator.py ├── object_detection_2d_geometric_ops.py ├── object_detection_2d_image_boxes_validation_utils.py ├── object_detection_2d_misc_utils.py ├── object_detection_2d_patch_sampling_ops.py └── object_detection_2d_photometric_ops.py ├── eval_utils ├── __init__.py ├── average_precision_evaluator.py └── coco_utils.py ├── examples ├── fish-bike.jpg ├── fish_bike.jpg ├── ssd300_pascalVOC_pred_01.png ├── ssd300_pascalVOC_pred_02.png ├── ssd300_pascalVOC_pred_03.png ├── ssd300_pascalVOC_pred_04.png ├── ssd300_pascalVOC_pred_05.png ├── ssd300_pascalVOC_pred_06.png ├── ssd300_pascalVOC_pred_07.png ├── ssd300_pascalVOC_pred_08.png ├── ssd300_pascalVOC_pred_09.png ├── ssd7_udacity_traffic_pred_01.png ├── ssd7_udacity_traffic_pred_02.png ├── ssd7_udacity_traffic_pred_03.png ├── ssd7_udacity_traffic_pred_04.png ├── ssd7_udacity_traffic_pred_05.png ├── trained_ssd300_pascalVOC2007_test_pred_01.png ├── trained_ssd300_pascalVOC2007_test_pred_01_no_gt.png ├── trained_ssd300_pascalVOC2007_test_pred_02.png ├── trained_ssd300_pascalVOC2007_test_pred_02_no_gt.png ├── trained_ssd300_pascalVOC2007_test_pred_03.png ├── trained_ssd300_pascalVOC2007_test_pred_03_no_gt.png ├── trained_ssd300_pascalVOC2007_test_pred_04.png ├── trained_ssd300_pascalVOC2007_test_pred_04_no_gt.png ├── trained_ssd300_pascalVOC2007_test_pred_05.png ├── trained_ssd300_pascalVOC2007_test_pred_05_no_gt.png └── trained_ssd300_pascalVOC2007_test_pred_06.png ├── keras_layers ├── __init__.py ├── keras_layer_AnchorBoxes.py ├── keras_layer_DecodeDetections.py ├── keras_layer_DecodeDetectionsFast.py └── keras_layer_L2Normalization.py ├── keras_loss_function ├── __init__.py └── keras_ssd_loss.py ├── misc_utils ├── __init__.py └── tensor_sampling_utils.py ├── models ├── __init__.py ├── keras_ssd300.py ├── keras_ssd512.py └── keras_ssd7.py ├── ssd300_evaluation.ipynb ├── ssd300_evaluation_COCO.ipynb ├── ssd300_inference.ipynb ├── ssd300_training.ipynb ├── ssd512_inference.ipynb ├── ssd7_training.ipynb ├── ssd_encoder_decoder ├── __init__.py ├── matching_utils.py ├── ssd_input_encoder.py └── ssd_output_decoder.py ├── training_summaries ├── ssd300_pascal_07+12_loss_history.png └── ssd300_pascal_07+12_training_summary.md └── weight_sampling_tutorial.ipynb /.gitattributes: -------------------------------------------------------------------------------- 1 | *.ipynb linguist-language=Python 2 | -------------------------------------------------------------------------------- /.github/stale.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/.github/stale.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /ISSUE_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/ISSUE_TEMPLATE.md -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bounding_box_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bounding_box_utils/bounding_box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/bounding_box_utils/bounding_box_utils.py -------------------------------------------------------------------------------- /data_generator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_generator/data_augmentation_chain_constant_input_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/data_generator/data_augmentation_chain_constant_input_size.py -------------------------------------------------------------------------------- /data_generator/data_augmentation_chain_original_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/data_generator/data_augmentation_chain_original_ssd.py -------------------------------------------------------------------------------- /data_generator/data_augmentation_chain_satellite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/data_generator/data_augmentation_chain_satellite.py -------------------------------------------------------------------------------- /data_generator/data_augmentation_chain_variable_input_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/data_generator/data_augmentation_chain_variable_input_size.py -------------------------------------------------------------------------------- /data_generator/object_detection_2d_data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/data_generator/object_detection_2d_data_generator.py -------------------------------------------------------------------------------- /data_generator/object_detection_2d_geometric_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/data_generator/object_detection_2d_geometric_ops.py -------------------------------------------------------------------------------- /data_generator/object_detection_2d_image_boxes_validation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/data_generator/object_detection_2d_image_boxes_validation_utils.py -------------------------------------------------------------------------------- /data_generator/object_detection_2d_misc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/data_generator/object_detection_2d_misc_utils.py -------------------------------------------------------------------------------- /data_generator/object_detection_2d_patch_sampling_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/data_generator/object_detection_2d_patch_sampling_ops.py -------------------------------------------------------------------------------- /data_generator/object_detection_2d_photometric_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/data_generator/object_detection_2d_photometric_ops.py -------------------------------------------------------------------------------- /eval_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval_utils/average_precision_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/eval_utils/average_precision_evaluator.py -------------------------------------------------------------------------------- /eval_utils/coco_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/eval_utils/coco_utils.py -------------------------------------------------------------------------------- /examples/fish-bike.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/fish-bike.jpg -------------------------------------------------------------------------------- /examples/fish_bike.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/fish_bike.jpg -------------------------------------------------------------------------------- /examples/ssd300_pascalVOC_pred_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd300_pascalVOC_pred_01.png -------------------------------------------------------------------------------- /examples/ssd300_pascalVOC_pred_02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd300_pascalVOC_pred_02.png -------------------------------------------------------------------------------- /examples/ssd300_pascalVOC_pred_03.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd300_pascalVOC_pred_03.png -------------------------------------------------------------------------------- /examples/ssd300_pascalVOC_pred_04.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd300_pascalVOC_pred_04.png -------------------------------------------------------------------------------- /examples/ssd300_pascalVOC_pred_05.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd300_pascalVOC_pred_05.png -------------------------------------------------------------------------------- /examples/ssd300_pascalVOC_pred_06.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd300_pascalVOC_pred_06.png -------------------------------------------------------------------------------- /examples/ssd300_pascalVOC_pred_07.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd300_pascalVOC_pred_07.png -------------------------------------------------------------------------------- /examples/ssd300_pascalVOC_pred_08.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd300_pascalVOC_pred_08.png -------------------------------------------------------------------------------- /examples/ssd300_pascalVOC_pred_09.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd300_pascalVOC_pred_09.png -------------------------------------------------------------------------------- /examples/ssd7_udacity_traffic_pred_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd7_udacity_traffic_pred_01.png -------------------------------------------------------------------------------- /examples/ssd7_udacity_traffic_pred_02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd7_udacity_traffic_pred_02.png -------------------------------------------------------------------------------- /examples/ssd7_udacity_traffic_pred_03.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd7_udacity_traffic_pred_03.png -------------------------------------------------------------------------------- /examples/ssd7_udacity_traffic_pred_04.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd7_udacity_traffic_pred_04.png -------------------------------------------------------------------------------- /examples/ssd7_udacity_traffic_pred_05.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/ssd7_udacity_traffic_pred_05.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_01.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_01_no_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_01_no_gt.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_02.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_02_no_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_02_no_gt.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_03.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_03.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_03_no_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_03_no_gt.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_04.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_04.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_04_no_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_04_no_gt.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_05.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_05.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_05_no_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_05_no_gt.png -------------------------------------------------------------------------------- /examples/trained_ssd300_pascalVOC2007_test_pred_06.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/examples/trained_ssd300_pascalVOC2007_test_pred_06.png -------------------------------------------------------------------------------- /keras_layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /keras_layers/keras_layer_AnchorBoxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/keras_layers/keras_layer_AnchorBoxes.py -------------------------------------------------------------------------------- /keras_layers/keras_layer_DecodeDetections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/keras_layers/keras_layer_DecodeDetections.py -------------------------------------------------------------------------------- /keras_layers/keras_layer_DecodeDetectionsFast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/keras_layers/keras_layer_DecodeDetectionsFast.py -------------------------------------------------------------------------------- /keras_layers/keras_layer_L2Normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/keras_layers/keras_layer_L2Normalization.py -------------------------------------------------------------------------------- /keras_loss_function/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /keras_loss_function/keras_ssd_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/keras_loss_function/keras_ssd_loss.py -------------------------------------------------------------------------------- /misc_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /misc_utils/tensor_sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/misc_utils/tensor_sampling_utils.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/keras_ssd300.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/models/keras_ssd300.py -------------------------------------------------------------------------------- /models/keras_ssd512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/models/keras_ssd512.py -------------------------------------------------------------------------------- /models/keras_ssd7.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/models/keras_ssd7.py -------------------------------------------------------------------------------- /ssd300_evaluation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/ssd300_evaluation.ipynb -------------------------------------------------------------------------------- /ssd300_evaluation_COCO.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/ssd300_evaluation_COCO.ipynb -------------------------------------------------------------------------------- /ssd300_inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/ssd300_inference.ipynb -------------------------------------------------------------------------------- /ssd300_training.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/ssd300_training.ipynb -------------------------------------------------------------------------------- /ssd512_inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/ssd512_inference.ipynb -------------------------------------------------------------------------------- /ssd7_training.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/ssd7_training.ipynb -------------------------------------------------------------------------------- /ssd_encoder_decoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ssd_encoder_decoder/matching_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/ssd_encoder_decoder/matching_utils.py -------------------------------------------------------------------------------- /ssd_encoder_decoder/ssd_input_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/ssd_encoder_decoder/ssd_input_encoder.py -------------------------------------------------------------------------------- /ssd_encoder_decoder/ssd_output_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/ssd_encoder_decoder/ssd_output_decoder.py -------------------------------------------------------------------------------- /training_summaries/ssd300_pascal_07+12_loss_history.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/training_summaries/ssd300_pascal_07+12_loss_history.png -------------------------------------------------------------------------------- /training_summaries/ssd300_pascal_07+12_training_summary.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/training_summaries/ssd300_pascal_07+12_training_summary.md -------------------------------------------------------------------------------- /weight_sampling_tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pierluigiferrari/ssd_keras/HEAD/weight_sampling_tutorial.ipynb --------------------------------------------------------------------------------