3.9 KiB
3.9 KiB
mmdet 模型支持列表
mmdet 是基于 pytorch 的检测工具箱，属于 OpenMMLab。
安装 mmdet
请参照 get_started.md 。
支持列表
| 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 |