mmdet_modelzoo.md 3.6 KB

Pre-trained Detection Models

人体 Bounding Box 检测模型

MMDetection 提供了基于 COCO 的包括 person 在内的 80 个类别的预训练模型,用户可前往 MMDetection Model Zoo 下载并将其用作人体 bounding box 识别模型。

手部 Bounding Box 检测模型

对于手部 bounding box 检测模型,我们提供了一个通过 MMDetection 基于 OneHand10K 数据库训练的模型。

基于 OneHand10K 测试集的测试结果

Arch Box AP ckpt log
Cascade_R-CNN X-101-64x4d-FPN-1class 0.817 ckpt log

脸部 Bounding Box 检测模型

对于脸部 bounding box 检测模型,我们提供了一个通过 MMDetection 基于 COCO-Face 数据库训练的 YOLOX 检测器。

基于 COCO-face 测试集的测试结果

Arch Box AP ckpt
YOLOX-s 0.408 ckpt

动物 Bounding Box 检测模型

COCO animals

COCO 数据集内包括了 10 种常见的 animal 类型:

(14: 'bird', 15: 'cat', 16: 'dog', 17: 'horse', 18: 'sheep', 19: 'cow', 20: 'elephant', 21: 'bear', 22: 'zebra', 23: 'giraffe') 。

用户如果需要使用以上类别的动物检测模型,可以前往 MMDetection Model Zoo 下载。

基于 MacaquePose 测试集的测试结果

Arch Box AP ckpt log
Faster_R-CNN_Res50-FPN-1class 0.840 ckpt log
Cascade_R-CNN X-101-64x4d-FPN-1class 0.879 ckpt log