update readme

ShiqiYu · Apr 28, 2024 · dc8fc2f · dc8fc2f
1 parent eb0b102
commit dc8fc2f
Show file tree

Hide file tree

Showing 4 changed files with 59 additions and 50 deletions.
diff --git a/README.md b/README.md
@@ -12,14 +12,16 @@ OpenGait is a flexible and extensible gait recognition project provided by the [
 The corresponding [paper](https://openaccess.thecvf.com/content/CVPR2023/papers/Fan_OpenGait_Revisiting_Gait_Recognition_Towards_Better_Practicality_CVPR_2023_paper.pdf) has been accepted by CVPR2023 as a highlight paper. 
 
 ## What's New
+- **[Apr 2024]** 
+Our team's latest checkpoints for projects such as DeepGaitv2, SkeletonGait, SkeletonGait++, and SwinGait will be released on [Hugging Face](https://huggingface.co/opengait/OpenGait). Additionally, previously released checkpoints will also be gradually made available on it.
 - **[Mar 2024]** [Chao](https://chaofan996.github.io) gives a talk about 'Progress in Gait Recognition'. The [video](https://event.baai.ac.cn/activities/768) and [slides](https://github.com/ChaoFan996/ChaoFan996.github.io/blob/main/240315-Progress%20in%20Gait%20Recognition.pdf) are both available😊
 - **[Mar 2024]** The code of [SkeletonGait++](https://arxiv.org/pdf/2311.13444.pdf) is released [here](opengait/modeling/models/skeletongait%2B%2B.py), and you can refer to [readme](configs/skeletongait) for details.
 - **[Mar 2024]** [BigGait](https://arxiv.org/pdf/2402.19122.pdf) has been accepted to CVPR2024🎉 Congratulations to [Dingqiang](https://faculty.sustech.edu.cn/?p=161709&tagid=yusq&cat=2&iscss=1&snapid=1&go=1&orderby=date)! This is his FIRST paper!
 - **[Jan 2024]** The code of transfomer-based [SwinGait](https://arxiv.org/pdf/2303.03301.pdf) is available at [here](opengait/modeling/models/swingait.py).
 - **[Dec 2023]** A new state-of-the-art baseline, i.e., [DeepGaitV2](https://arxiv.org/pdf/2303.03301.pdf), is available at [here](opengait/modeling/models/deepgaitv2.py)! 
 - [Nov 2023] The first million-level unlabeled gait dataset, i.e., [GaitLU-1M](https://ieeexplore.ieee.org/document/10242019), is released and supported in [datasets/GaitLU-1M](datasets/GaitLU-1M/README.md).
 - [Oct 2023] Several representative pose-based methods are supported in [opengait/modeling/models](./opengait/modeling/models). This feature is mainly inherited from [FastPoseGait](https://github.com/BNU-IVC/FastPoseGait). Many thanks to the contributors😊.
-- [July 2023] [CCPG](https://github.com/BNU-IVC/CCPG) is supported in [datasets/CCPG](./datasets/CCPG).
+<!-- - [July 2023] [CCPG](https://github.com/BNU-IVC/CCPG) is supported in [datasets/CCPG](./datasets/CCPG). -->
 <!-- - - - [July 2023] [SUSTech1K](https://lidargait.github.io) is released and supported in [datasets/SUSTech1K](./datasets/SUSTech1K).
 [May 2023] A real gait recognition system [All-in-One-Gait](https://github.com/jdyjjj/All-in-One-Gait) provided by [Dongyang Jin](https://github.com/jdyjjj) is available.
 [Apr 2023] [CASIA-E](datasets/CASIA-E/README.md) is supported by OpenGait.

diff --git a/configs/swingait/swingait3D_B1122C2_SUSTech1K.yaml b/configs/swingait/swingait3D_B1122C2_SUSTech1K.yaml
@@ -82,8 +82,9 @@ trainer_cfg:
     batch_size:
       - 8 # TripletSampler, batch_size[0] indicates Number of Identity
       - 8 #                 batch_size[1] indicates Samples sequqnce for each Identity
-    frames_num_fixed: 10 # fixed frames number for training
-    sample_type: fixed_unordered # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered
+    frames_num_fixed: 30 # fixed frames number for training
+    frames_skip_num: 4
+    sample_type: fixed_ordered # # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered
     type: TripletSampler
   transform:
     - type: Compose

diff --git a/docs/1.1.skeleton_model_zoo.md b/docs/1.1.skeleton_model_zoo.md
@@ -8,6 +8,7 @@
 |                                                        [GaitGraph1](https://ieeexplore.ieee.org/document/9506717)                                                        |   AlphaPose    |           2.81            |           4.25           |[gaitgraph1.yaml](../configs/gaitgraph1/gaitgraph1_phase1_OUMVLP.yaml),[gaitgraph1_17.yaml](../configs/gaitgraph1/gaitgraph1_phase1_OUMVLP17.yaml) |[alphapose format](https://huggingface.co/opengait/OpenGait/tree/main/OUMVLP/GaitGraph1/GaitGraph1_phase1), [coco2017 format](https://huggingface.co/opengait/OpenGait/tree/main/OUMVLP/GaitGraph1/GaitGraph1_phase1_17)|
 | [GaitGraph2](https://openaccess.thecvf.com/content/CVPR2022W/Biometrics/papers/Teepe_Towards_a_Deeper_Understanding_of_Skeleton-Based_Gait_Recognition_CVPRW_2022_paper) |   AlphaPose    |           62.10           |          70.75           |[gaitgraph2.yaml](../configs/gaitgraph2/gaitgraph2_OUMVLP.yaml),[gaitgraph2_17.yaml](../configs/gaitgraph2/gaitgraph2_OUMVLP17.yaml) |[alphapose format](https://huggingface.co/opengait/OpenGait/tree/main/OUMVLP/GaitGraph2/GaitGraph2_phase2), [coco2017 format](https://huggingface.co/opengait/OpenGait/tree/main/OUMVLP/GaitGraph2/GaitGraph2_phase2_17)|
 |                                                                [GaitTR](https://arxiv.org/abs/2204.03873)                                                                |   AlphaPose    |           40.31           |          39.78           |[gaittr.yaml](../configs/gaittr/gaittr_OUMVLP.yaml),[gaittr_17.yaml](../configs/gaittr/gaittr_OUMVLP17.yaml) |[alphapose format](https://huggingface.co/opengait/OpenGait/tree/main/OUMVLP/GaitTR/GaitTR), [coco2017 format](https://huggingface.co/opengait/OpenGait/tree/main/OUMVLP/GaitTR/GaitTR)|
+| [SkeletonGait(AAAI2024)](https://ojs.aaai.org/index.php/AAAI/article/view/27933/27887) | AlphaPose | - | 67.4 |[skeletongait_OUMVLP.yaml](../configs/skeletongait/skeletongait_OUMVLP.yaml) |[model](https://huggingface.co/opengait/OpenGait/blob/main/OUMVLP/DeepGaitV2/SkeletonGait/checkpoints/SkeletonGait-120000.pt)|
 
 
 
@@ -18,9 +19,10 @@
 
 |                            Model                             | Pose Estimator | Rank-1 |Configuration| Download checkpoint |
 | :----------------------------------------------------------: | :------------: |:------:|:------:|:------:|
-|  [GaitGraph1](https://ieeexplore.ieee.org/document/9506717)  |     HRNet      |  9.82  |[gaitgraph1.yaml](../configs/gaitgraph1/gaitgraph1_phase1_GREW.yaml)|https://huggingface.co/opengait/OpenGait/tree/main/GREW/GaitGraph1/GaitGraph1_phase1|
-| [GaitGraph2](https://openaccess.thecvf.com/content/CVPR2022W/Biometrics/papers/Teepe_Towards_a_Deeper_Understanding_of_Skeleton-Based_Gait_Recognition_CVPRW_2022_paper) |     HRNet      | 34.61 | [gaitgraph2.yaml](../configs/gaitgraph2/gaitgraph2_GREW.yaml)|https://huggingface.co/opengait/OpenGait/tree/main/GREW/GaitGraph2/GaitGraph2|
-|          [GaitTR](https://arxiv.org/abs/2204.03873)          |     HRNet      | 48.60  | [gaittr.yaml](../configs/gaittr/gaittr_GREW.yaml)|https://huggingface.co/opengait/OpenGait/tree/main/GREW/GaitTR/GaitTR|
+|  [GaitGraph1](https://ieeexplore.ieee.org/document/9506717)  |     HRNet      |  9.82  |[gaitgraph1.yaml](../configs/gaitgraph1/gaitgraph1_phase1_GREW.yaml)|[model](https://huggingface.co/opengait/OpenGait/tree/main/GREW/GaitGraph1/GaitGraph1_phase1)|
+| [GaitGraph2](https://openaccess.thecvf.com/content/CVPR2022W/Biometrics/papers/Teepe_Towards_a_Deeper_Understanding_of_Skeleton-Based_Gait_Recognition_CVPRW_2022_paper) |     HRNet      | 34.61 | [gaitgraph2.yaml](../configs/gaitgraph2/gaitgraph2_GREW.yaml)|[model](https://huggingface.co/opengait/OpenGait/tree/main/GREW/GaitGraph2/GaitGraph2)|
+|          [GaitTR](https://arxiv.org/abs/2204.03873)          |     HRNet      | 48.60  | [gaittr.yaml](../configs/gaittr/gaittr_GREW.yaml)|[model](https://huggingface.co/opengait/OpenGait/tree/main/GREW/GaitTR/GaitTR)|
+| [SkeletonGait(AAAI2024)](https://ojs.aaai.org/index.php/AAAI/article/view/27933/27887) | HRNet | 77.4 | [skeletongait++_GREW.yaml](../configs/skeletongait/skeletongait++_GREW.yaml) |[model](https://huggingface.co/opengait/OpenGait/blob/main/GREW/DeepGaitV2/SkeletonGait/checkpoints/SkeletonGait-180000.pt)|
 
 
 
@@ -30,10 +32,10 @@
 
 |                            Model                             | Pose Estimator | Rank-1 |Configuration|Download checkpoint|
 | :----------------------------------------------------------: | :------------: |:------:|:------:|:------:|
-|  [GaitGraph1](https://ieeexplore.ieee.org/document/9506717)  |     HRNet      |  9.00  |[gaitgraph1.yaml](../configs/gaitgraph1/gaitgraph1_phase1_Gait3D.yaml)|https://huggingface.co/opengait/OpenGait/tree/main/Gait3D/GaitGraph1/GaitGraph1_phase1|
-| [GaitGraph2](https://openaccess.thecvf.com/content/CVPR2022W/Biometrics/papers/Teepe_Towards_a_Deeper_Understanding_of_Skeleton-Based_Gait_Recognition_CVPRW_2022_paper) |     HRNet      | 11.00  |[gaitgraph2.yaml](../configs/gaitgraph2/gaitgraph2_Gait3D.yaml)|https://huggingface.co/opengait/OpenGait/tree/main/Gait3D/GaitGraph2/GaitGraph2|
-|          [GaitTR](https://arxiv.org/abs/2204.03873)          |     HRNet      |  7.20  |[gaittr.yaml](../configs/gaittr/gaittr_Gait3D.yaml)|https://huggingface.co/opengait/OpenGait/tree/main/Gait3D/GaitTR/GaitTR|
-
+|  [GaitGraph1](https://ieeexplore.ieee.org/document/9506717)  |     HRNet      |  9.00  |[gaitgraph1.yaml](../configs/gaitgraph1/gaitgraph1_phase1_Gait3D.yaml)|[model](https://huggingface.co/opengait/OpenGait/tree/main/Gait3D/GaitGraph1/GaitGraph1_phase1)|
+| [GaitGraph2](https://openaccess.thecvf.com/content/CVPR2022W/Biometrics/papers/Teepe_Towards_a_Deeper_Understanding_of_Skeleton-Based_Gait_Recognition_CVPRW_2022_paper) |     HRNet      | 11.00  |[gaitgraph2.yaml](../configs/gaitgraph2/gaitgraph2_Gait3D.yaml)|[model](https://huggingface.co/opengait/OpenGait/tree/main/Gait3D/GaitGraph2/GaitGraph2)|
+|          [GaitTR](https://arxiv.org/abs/2204.03873)          |     HRNet      |  7.20  |[gaittr.yaml](../configs/gaittr/gaittr_Gait3D.yaml)|[model](https://huggingface.co/opengait/OpenGait/tree/main/Gait3D/GaitTR/GaitTR)|
+| [SkeletonGait(AAAI2024)](https://ojs.aaai.org/index.php/AAAI/article/view/27933/27887) | HRNet | 38.1 |[skeletongait_Gait3D.yaml](../configs/skeletongait/skeletongait_Gait3D.yaml)|[model](https://huggingface.co/opengait/OpenGait/blob/main/Gait3D/DeepGaitV2/SkeletonGait/checkpoints/SkeletonGait-60000.pt)|
 
 
 
@@ -42,4 +44,3 @@
 * We greatly appreciate the contributions of the author mentioned above.
 * We are very grateful for the contribution of the fastpositait team.
 
-