EasyCV/docs/source/model_zoo_pose.md

34 lines
4.3 KiB
Markdown
Raw Normal View History

# Pose/keypoints Model Zoo
## Body
**Topdown Heatmap + Hrnet + Udp on Coco**
| Config | Params | AP | AP50 | AP75 | AR | AR50 | Download |
| ---------------------------------------------------------------------------------------------------------------------- | ------ | ----- | ----- | ----- | ----- | ----- | -------------------------------------------------------------------------------------------------------------------- |
| [hrnet_w48_coco_256x192_udp](https://github.com/alibaba/EasyCV/blob/master/configs/pose/hrnet_w48_coco_256x192_udp.py) | 63.6M | 0.781 | 0.936 | 0.849 | 0.801 | 0.941 | [model](http://pai-vision-data-hz.oss-cn-zhangjiakou.aliyuncs.com/EasyCV/modelzoo/pose/top_down_hrnet/epoch_210.pth) |
**Topdown Heatmap + Litehrnet on Coco**
| Config | Params | AP | AP50 | AP75 | AR | AR50 | Download |
|:-------------------------------------------------------------------------------------------------------------------- | ------ | ----- | ----- | ----- | ----- | ----- | ------------------------------------------------------------------------------------------------------------------------ |
| [litehrnet_30_coco_384x288](https://github.com/alibaba/EasyCV/blob/master/configs/pose/litehrnet_30_coco_384x288.py) | 1.76M | 0.710 | 0.905 | 0.783 | 0.742 | 0.919 | [model](http://pai-vision-data-hz.oss-cn-zhangjiakou.aliyuncs.com/EasyCV/modelzoo/pose/top_down_litehrnet/epoch_210.pth) |
## WholeBody
**Topdown Heatmap + Hrnet + Dark on Coco-Wholebody**
| Config | Params | Body AP | Body AR | Foot AP | Foot AR | Face AP | Face AR | LeftHand AP | LeftHand AR | RightHand AP | RightHand AP | Whole AP | Whole AR | Download |
| ---------------------------------------------------------------------------------------------------------------------------------------------------------------- | ------ | ------- | ------- | ------- | ------- | ------- | ------- | ----------- | ----------- | ------------ | ------------ | -------- | -------- | --------------------------------------------------------------------------------------------------------------------- |
| [hrnet_w48_coco_wholebody_384x288_dark_plus](https://github.com/alibaba/EasyCV/blob/master/configs/pose/wholebody/hrnet_w48_coco_wholebody_384x288_dark_plus.py) | 63.6M | 0.766 | 0.806 | 0.746 | 0.805 | 0.856 | 0.892 | 0.631 | 0.694 | 0.595 | 0.667 | 0.684 | 0.742 | [model](http://pai-vision-data-hz.oss-cn-zhangjiakou.aliyuncs.com/EasyCV/modelzoo/pose/wholebody/hrnet/epoch_290.pth) |
Note: `plus` means the model is first pre-trained on original COCO dataset, and then fine-tuned on COCO-WholeBody dataset.
## Hand
**Topdown Heatmap + Hrnet + Dark + Coco + Wholebody on Coco_wholebody_hand**
| Config | Params | PCK | AUC | EPE | Download |
| ----------------------------------------------------------------------------------------------------------------------------------------------------------- | ------ | ----- | ----- | ---- | ------------------------------------------------------------------------------------------------------------------------ |
| [hrnet_w18_coco_wholebody_hand_256x256_dark](https://github.com/alibaba/EasyCV/blob/master/configs/pose/hand/hrnet_w18_coco_wholebody_hand_256x256_dark.py) | 9.6M | 0.816 | 0.839 | 4.39 | [model](http://pai-vision-data-hz.oss-cn-zhangjiakou.aliyuncs.com/EasyCV/modelzoo/pose/hand/hrnet/hrnet_w18_256x256.pth) |