mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
efd39953d7
* resolve conflicts * update ut and docs * fix ut * refine docstring * add comments and refine UT * resolve comments * resolve comments * update doc * add roll export * check backend * update regression test
3.8 KiB
3.8 KiB
MMDetection Support
MMDetection is an open source object detection toolbox based on PyTorch. It is a part of the OpenMMLab project.
MMDetection installation tutorial
Please refer to get_started.md for installation.
List of MMDetection models supported by MMDeploy
| 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 |
| 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 |