-
- Downloads
Object Detection Compression (#343)
Add an example of compressing OD pytorch models. In this example we compress torchvision's object detection models - FasterRCNN / MaskRCNN / KeypointRCNN. We've modified the reference code for object detection to allow easy compression scheduling with YAML configuration.
Showing
- distiller/model_transforms.py 2 additions, 1 deletiondistiller/model_transforms.py
- distiller/models/__init__.py 3 additions, 1 deletiondistiller/models/__init__.py
- distiller/quantization/sim_bn_fold.py 16 additions, 5 deletionsdistiller/quantization/sim_bn_fold.py
- examples/object_detection_compression/README.md 57 additions, 0 deletionsexamples/object_detection_compression/README.md
- examples/object_detection_compression/__init__.py 0 additions, 0 deletionsexamples/object_detection_compression/__init__.py
- examples/object_detection_compression/coco_eval.py 351 additions, 0 deletionsexamples/object_detection_compression/coco_eval.py
- examples/object_detection_compression/coco_utils.py 253 additions, 0 deletionsexamples/object_detection_compression/coco_utils.py
- examples/object_detection_compression/compress_detector.py 356 additions, 0 deletionsexamples/object_detection_compression/compress_detector.py
- examples/object_detection_compression/data/download_dataset.sh 9 additions, 0 deletions...les/object_detection_compression/data/download_dataset.sh
- examples/object_detection_compression/engine.py 125 additions, 0 deletionsexamples/object_detection_compression/engine.py
- examples/object_detection_compression/group_by_aspect_ratio.py 189 additions, 0 deletions...les/object_detection_compression/group_by_aspect_ratio.py
- examples/object_detection_compression/logging.conf 54 additions, 0 deletionsexamples/object_detection_compression/logging.conf
- examples/object_detection_compression/maskrcnn.scheduler_agp.non_parallel.yaml 210 additions, 0 deletions...tion_compression/maskrcnn.scheduler_agp.non_parallel.yaml
- examples/object_detection_compression/maskrcnn.scheduler_agp.yaml 267 additions, 0 deletions.../object_detection_compression/maskrcnn.scheduler_agp.yaml
- examples/object_detection_compression/requirements.txt 2 additions, 0 deletionsexamples/object_detection_compression/requirements.txt
- examples/object_detection_compression/transforms.py 52 additions, 0 deletionsexamples/object_detection_compression/transforms.py
- examples/object_detection_compression/utils.py 328 additions, 0 deletionsexamples/object_detection_compression/utils.py
Loading
Please register or sign in to comment