mirror of https://github.com/open-mmlab/mmpose
3.9 KiB
3.9 KiB
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 |
RTMDet-nano | 0.760 | ckpt | - |
脸部 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 |