MMDetection is an open source object detection toolbox based on PyTorch. It is a part of the OpenMMLab project.
Please refer to get_started.md for installation.
| Model | Task | OnnxRuntime | TensorRT | ncnn | PPLNN | OpenVINO | Model config |
|---|---|---|---|---|---|---|---|
| ATSS | ObjectDetection | Y | Y | N | N | Y | config |
| FCOS | ObjectDetection | Y | Y | Y | N | Y | config |
| FoveaBox | ObjectDetection | Y | N | N | N | Y | config |
| FSAF | ObjectDetection | Y | Y | Y | Y | Y | config |
| RetinaNet | ObjectDetection | Y | Y | Y | Y | Y | config |
| SSD | ObjectDetection | Y | Y | Y | N | Y | config |
| VFNet | ObjectDetection | N | N | N | N | Y | config |
| YOLOv3 | ObjectDetection | Y | Y | Y | N | Y | config |
| YOLOX | ObjectDetection | Y | Y | Y | N | Y | config |
| Cascade R-CNN | ObjectDetection | Y | Y | N | Y | Y | config |
| Faster R-CNN | ObjectDetection | Y | Y | Y | Y | Y | config |
| Faster R-CNN + DCN | ObjectDetection | Y | Y | Y | Y | Y | config |
| GFL | ObjectDetection | Y | Y | N | ? | Y | config |
| RepPoints | ObjectDetection | N | Y | N | ? | Y | config |
| DETR | ObjectDetection | Y | Y | N | ? | Y | config |
| Cascade Mask R-CNN | InstanceSegmentation | Y | N | N | N | Y | config |
| Mask R-CNN | InstanceSegmentation | Y | Y | N | N | Y | config |
| Swin Transformer | InstanceSegmentation | Y | Y | N | N | N | config |