update litehrnet modelzoo (open-mmlab#1120)

wjkim81 · Jan 5, 2022 · a53cae9 · a53cae9
1 parent 7186a03
commit a53cae9
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 1 deletion.
diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_coco.md b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_coco.md
@@ -36,5 +36,7 @@ Results on COCO val2017 with detector having human AP of 56.4 on COCO val2017 da
 
 | Arch | Input Size | AP | AP<sup>50</sup> | AP<sup>75</sup> | AR | AR<sup>50</sup> | ckpt | log |
 | :----------------- | :-----------: | :------: | :------: | :------: | :------: | :------: |:------: |:------: |
+| [LiteHRNet-18](/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_18_coco_256x192.py) | 256x192 | 0.643 | 0.868 | 0.720 | 0.706 | 0.912 | [ckpt](https://download.openmmlab.com/mmpose/top_down/litehrnet/litehrnet18_coco_256x192-6bace359_20211230.pth) | [log](https://download.openmmlab.com/mmpose/top_down/litehrnet/litehrnet18_coco_256x192_20211230.log.json) |
+| [LiteHRNet-18](/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_18_coco_384x288.py) | 384x288 | 0.677 | 0.878 | 0.746 | 0.735 | 0.920 | [ckpt](https://download.openmmlab.com/mmpose/top_down/litehrnet/litehrnet18_coco_384x288-8d4dac48_20211230.pth) | [log](https://download.openmmlab.com/mmpose/top_down/litehrnet/litehrnet18_coco_384x288_20211230.log.json) |
 | [LiteHRNet-30](/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_30_coco_256x192.py) | 256x192 | 0.675 | 0.881 | 0.754 | 0.736 | 0.924 | [ckpt](https://download.openmmlab.com/mmpose/top_down/litehrnet/litehrnet30_coco_256x192-4176555b_20210626.pth) | [log](https://download.openmmlab.com/mmpose/top_down/litehrnet/litehrnet30_coco_256x192_20210626.log.json) |
 | [LiteHRNet-30](/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_30_coco_384x288.py) | 384x288 | 0.700 | 0.884 | 0.776 | 0.758 | 0.928 | [ckpt](https://download.openmmlab.com/mmpose/top_down/litehrnet/litehrnet30_coco_384x288-a3aef5c4_20210626.pth) | [log](https://download.openmmlab.com/mmpose/top_down/litehrnet/litehrnet30_coco_384x288_20210626.log.json) |
diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_coco.yml b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_coco.yml
@@ -5,12 +5,46 @@ Collections:
  URL: https://arxiv.org/abs/2104.06403
  README: https://github.com/open-mmlab/mmpose/blob/master/docs/en/papers/algorithms/litehrnet.md
 Models:
-- Config: configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_30_coco_256x192.py
+- Config: configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_18_coco_256x192.py
  In Collection: LiteHRNet
  Metadata:
  Architecture: &id001
  - LiteHRNet
  Training Data: COCO
+ Name: topdown_heatmap_litehrnet_18_coco_256x192
+ README: configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_coco.md
+ Results:
+ - Dataset: COCO
+ Metrics:
+ AP: 0.643
+ AP@0.5: 0.868
+ AP@0.75: 0.72
+ AR: 0.706
+ AR@0.5: 0.912
+ Task: Body 2D Keypoint
+ Weights: https://download.openmmlab.com/mmpose/top_down/litehrnet/litehrnet18_coco_256x192-6bace359_20211230.pth
+- Config: configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_18_coco_384x288.py
+ In Collection: LiteHRNet
+ Metadata:
+ Architecture: *id001
+ Training Data: COCO
+ Name: topdown_heatmap_litehrnet_18_coco_384x288
+ README: configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_coco.md
+ Results:
+ - Dataset: COCO
+ Metrics:
+ AP: 0.677
+ AP@0.5: 0.878
+ AP@0.75: 0.746
+ AR: 0.735
+ AR@0.5: 0.92
+ Task: Body 2D Keypoint
+ Weights: https://download.openmmlab.com/mmpose/top_down/litehrnet/litehrnet18_coco_384x288-8d4dac48_20211230.pth
+- Config: configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/litehrnet_30_coco_256x192.py
+ In Collection: LiteHRNet
+ Metadata:
+ Architecture: *id001
+ Training Data: COCO
  Name: topdown_heatmap_litehrnet_30_coco_256x192
  Results:
  - Dataset: COCO