From 929584632e1fd90769bd548a90e722a89004577f Mon Sep 17 00:00:00 2001 From: Yufei Date: Fri, 6 Jan 2023 12:30:28 +0800 Subject: [PATCH] bug fix in MoEMlp input params bug fix in MoEMlp input params remove the extra head update readme for ViTPose+ --- README.md | 150 ++++-- .../ap10k/ViTPose_base_ap10k_256x192.py | 157 ++++++ .../ap10k/ViTPose_huge_ap10k_256x192.py | 157 ++++++ .../ap10k/ViTPose_large_ap10k_256x192.py | 157 ++++++ .../ap10k/ViTPose_small_ap10k_256x192.py | 157 ++++++ .../apt36k/ViTPose_base_apt36k_256x192.py | 157 ++++++ .../apt36k/ViTPose_huge_ap36k_256x192.py | 157 ++++++ .../apt36k/ViTPose_large_apt36k_256x192.py | 157 ++++++ .../apt36k/ViTPose_small_apt36k_256x192.py | 157 ++++++ .../aic/ViTPose_small_aic_256x192.py | 151 ++++++ .../coco/ViTPose_small_coco_256x192.py | 170 ++++++ .../coco/ViTPose_small_simple_coco_256x192.py | 170 ++++++ ...mpii+ap10k+apt36k+wholebody_256x192_udp.py | 9 - ...mpii+ap10k+apt36k+wholebody_256x192_udp.py | 9 - ...mpii+ap10k+apt36k+wholebody_256x192_udp.py | 9 - ...mpii+ap10k+apt36k+wholebody_256x192_udp.py | 491 ++++++++++++++++++ .../mpii/ViTPose_small_mpii_256x192.py | 146 ++++++ .../ochuman/ViTPose_small_ochuman_256x192.py | 153 ++++++ .../ViTPose_base_interhand2d_all_256x192.py | 162 ++++++ .../ViTPose_huge_interhand2d_all_256x192.py | 162 ++++++ .../ViTPose_large_interhand2d_all_256x192.py | 162 ++++++ .../ViTPose_small_interhand2d_all_256x192.py | 162 ++++++ .../ViTPose_base_wholebody_256x192.py | 149 ++++++ .../ViTPose_huge_wholebody_256x192.py | 149 ++++++ .../ViTPose_large_wholebody_256x192.py | 149 ++++++ .../ViTPose_small_wholebody_256x192.py | 149 ++++++ figures/Throughput.png | Bin 119184 -> 105718 bytes mmpose/models/backbones/vit_moe.py | 3 +- tools/model_split.py | 93 ++++ 29 files changed, 3992 insertions(+), 62 deletions(-) create mode 100644 configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/ViTPose_base_ap10k_256x192.py create mode 100644 configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/ViTPose_huge_ap10k_256x192.py create mode 100644 configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/ViTPose_large_ap10k_256x192.py create mode 100644 configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/ViTPose_small_ap10k_256x192.py create mode 100644 configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_base_apt36k_256x192.py create mode 100644 configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_huge_ap36k_256x192.py create mode 100644 configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_large_apt36k_256x192.py create mode 100644 configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_small_apt36k_256x192.py create mode 100644 configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/ViTPose_small_aic_256x192.py create mode 100644 configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_small_coco_256x192.py create mode 100644 configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_small_simple_coco_256x192.py create mode 100644 configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/vitPose+_small_coco+aic+mpii+ap10k+apt36k+wholebody_256x192_udp.py create mode 100644 configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/ViTPose_small_mpii_256x192.py create mode 100644 configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/ViTPose_small_ochuman_256x192.py create mode 100644 configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/ViTPose_base_interhand2d_all_256x192.py create mode 100644 configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/ViTPose_huge_interhand2d_all_256x192.py create mode 100644 configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/ViTPose_large_interhand2d_all_256x192.py create mode 100644 configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/ViTPose_small_interhand2d_all_256x192.py create mode 100644 configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/ViTPose_base_wholebody_256x192.py create mode 100644 configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/ViTPose_huge_wholebody_256x192.py create mode 100644 configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/ViTPose_large_wholebody_256x192.py create mode 100644 configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/ViTPose_small_wholebody_256x192.py create mode 100644 tools/model_split.py diff --git a/README.md b/README.md index 97c75fd..a7cfd79 100644 --- a/README.md +++ b/README.md @@ -20,7 +20,7 @@