openvinotoolkit / mmdetection

OpenVINO Training Extensions Object Detection
https://github.com/opencv/openvino_training_extensions
Other
93 stars 31 forks source link
atss cascade-rcnn compression computer-vision deep-learning faster-rcnn fcos instance-segmentation mask-rcnn nncf object-detection onnx onnxruntime openvino pytorch quantization-aware-training retinanet sparsity ssd

MMDetection

This is an Object Detection and Instance Segmentation toolbox, that is a part of OpenVINO™ Training Extensions.

Project is based on open-mmlab's MMDetection.

Installation

Please refer to install.md for installation and dataset preparation.

Get Started

Please see getting_started.md for the basic usage. There are also tutorials for finetuning models, adding new dataset, designing data pipeline, and adding new modules.

Models compression

This project allows making compression of the models by NNCF framework.

See details in nncf_compression.md.

Model Zoo

Please see model_zoo.md for the list of pre-trained models.

Contributing

We appreciate all contributions to improve this project. Please refer to CONTRIBUTING.md for the contributing guideline.

Contact

This repo is currently maintained by Ilya Krylov (@Ilya-Krylov), Pavel Druzhkov (@druzhkov-paul) and Dmitry Sidnev (@DmitriySidnev).