mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
ba5351e209
* add gfl_trt * add gfl_head.py * add batch_integral * lint code * add gfl unit test * fix unit test * add gfl benchmark * fix unit test bug * Update gfl_head.py * Update __init__.py remove '**_forward_single' * fix lint error and ut error * fix docs and benchmark Co-authored-by: VVsssssk <shenkun@pjlab.org.cn>
3.5 KiB
3.5 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 |
| Cascade Mask R-CNN | InstanceSegmentation | Y | N | N | N | Y | config |
| Mask R-CNN | InstanceSegmentation | Y | Y | N | N | Y | config |
Reminder
None
FAQs
None