|
--- |
|
license: openrail |
|
--- |
|
Pretrained models of our method **MultiAugs** |
|
|
|
Title: *Boosting Semi-Supervised 2D Human Pose Estimation by Revisiting Data Augmentation and Consistency Training* |
|
|
|
Paper link: https://arxiv.org/abs/2402.11566 |
|
|
|
Code link: https://github.com/hnuzhy/MultiAugs |
|
|
|
**COCO1K / COCO5K / COCO10K** |
|
|
|
*trained on `partly labeled (1k, 5k or 10k) COCO train-set` and `left unlabeled COCO train-set`* |
|
* **ResNet-18 (Pose_Cons using single network)(256x192, COCO1K, 30 epochs)**: [pose_cons_18-COCO1K_e30-model_best.pth.tar](./pose_cons_18-COCO1K_e30-model_best.pth.tar) |
|
* **ResNet-18 (Pose_Cons using single network)(256x192, COCO5K, 70 epochs)**: [pose_cons_18-COCO5K_e70-model_best.pth.tar](./pose_cons_18-COCO5K_e70-model_best.pth.tar) |
|
* **ResNet-18 (Pose_Cons using single network)(256x192, COCO10K, 100 epochs)**: [pose_cons_18-COCO10K_e100-model_best.pth.tar](./pose_cons_18-COCO10K_e100-model_best.pth.tar) |
|
* **ResNet-18 (Pose_Dual using dual networks)(256x192, COCO1K, 30 epochs)**: [pose_dual_18-COCO1K_e30-model_best.pth.tar](./pose_dual_18-COCO1K_e30-model_best.pth.tar) |
|
* **ResNet-18 (Pose_Dual using dual networks)(256x192, COCO5K, 70 epochs)**: [pose_dual_18-COCO5K_e70-model_best.pth.tar](./pose_dual_18-COCO5K_e70-model_best.pth.tar) |
|
* **ResNet-18 (Pose_Dual using dual networks)(256x192, COCO10K, 100 epochs)**: [pose_dual_18-COCO10K_e100-model_best.pth.tar](./pose_dual_18-COCO10K_e100-model_best.pth.tar) |
|
|
|
|
|
**COCOall + COCOunlabel** |
|
|
|
*trained on `labeled COCO train-set` and `unlabeled COCO unlabeled-set`* |
|
* **ResNet-50 (Pose_Cons) (256x192, 400 epochs)**: [pose_cons_50-COCO_COCOunlabel_e400-model_best.pth.tar](./pose_cons_50-COCO_COCOunlabel_e400-model_best.pth.tar) |
|
* **ResNet-101 (Pose_Cons) (256x192, 400 epochs)**: [pose_cons_101-COCO_COCOunlabel_e400-model_best.pth.tar](./pose_cons_101-COCO_COCOunlabel_e400-model_best.pth.tar) |
|
* **HRNet-w48 (Pose_Cons) (384x288, 300 epochs)**: [pose_cons_w48-COCO_COCOunlabel_e300-model_best.pth.tar](./pose_cons_w48-COCO_COCOunlabel_e300-model_best.pth.tar) |
|
* **ResNet-50 (Pose_Dual) (256x192, 400 epochs)**: [pose_dual_50-COCO_COCOunlabel_e400-model_best.pth.tar](./pose_dual_50-COCO_COCOunlabel_e400-model_best.pth.tar) |
|
* **ResNet-101 (Pose_Dual) (256x192, 400 epochs)**: [pose_dual_101-COCO_COCOunlabel_e400-model_best.pth.tar](./pose_dual_101-COCO_COCOunlabel_e400-model_best.pth.tar) |
|
* **HRNet-w48 (Pose_Dual) (384x288, 300 epochs)**: [pose_dual_w48-COCO_COCOunlabel_e300-model_best.pth.tar](./pose_dual_w48-COCO_COCOunlabel_e300-model_best.pth.tar) |
|
|
|
|
|
**MPII + AIC** |
|
|
|
*trained on `labeled MPII train-set` and `unlabeled AIC train-set`* |
|
* **HRNet-w32 (Pose_Dual) (256x256, 400 epochs)** [*We are sorry that it cannot be released due to company copyright issues*] |
|
|
|
|
|
|