GREW
收藏Gait Recognition in the Wild: A Large-scale Benchmark and NAS-based Baseline
数据集准备
训练与评估
1. 超网络训练
bash CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -u -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs configs/sposgait/sposgait_large_GREW_supertraining_triplet.yaml --phase train
python -m torch.distributed.launch: DDP 启动指令。--nproc_per_node: 使用的GPU数量,必须等于CUDA_VISIBLE_DEVICES的长度。--cfgs: 配置文件路径。--phase: 指定为train。--log_to_file: 如果指定,终端日志将同时写入磁盘。
2. 搜索
bash CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -u -m torch.distributed.launch --nproc_per_node=8 opengait/search.py --cfgs ./configs/sposgait/sposgait_large_GREW_supertraining_triplet.yaml --max-epochs 20
3. 重新训练
bash CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -u -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./configs/sposgait/retrain/sposgait_large_GREW-train20000id_retrain.yaml --phase train
python -m torch.distributed.launch: DDP 启动指令。--nproc_per_node: 使用的GPU数量,必须等于CUDA_VISIBLE_DEVICES的长度。--cfgs: 配置文件路径。--phase: 指定为train。--log_to_file: 如果指定,终端日志将同时写入磁盘。
4. 测试
bash CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./configs/sposgait/retrain/sposgait_large_GREW-train20000id_retrain.yaml --phase test
--phase: 指定为test。--iter: 指定迭代检查点。
计算FLOPs和参数
bash CUDA_VISIBLE_DEVICES=0 python -u -m torch.distributed.launch --nproc_per_node=1 opengait/calculate_flops_and_params.py --cfgs configs/sposgait/retrain/sposgait_large_GREW-train20000id_retrain.yaml
引用
如果此工作对您的研究有帮助,请考虑引用以下BibTeX条目: bibtex @inproceedings{zhu2021gait, title={Gait recognition in the wild: A benchmark}, author={Zhu, Zheng and Guo, Xianda and Yang, Tian and Huang, Junjie and Deng, Jiankang and Huang, Guan and Du, Dalong and Lu, Jiwen and Zhou, Jie}, booktitle={Proceedings of the IEEE/CVF international conference on computer vision}, pages={14789--14799}, year={2021} } @article{guo2022gait, title={Gait Recognition in the Wild: A Large-scale Benchmark and NAS-based Baseline}, author={Guo, Xianda and Zhu, Zheng and Yang, Tian and Lin, Beibei and Huang, Junjie and Deng, Jiankang and Huang, Guan and Zhou, Jie and Lu, Jiwen}, journal={arXiv e-prints}, pages={arXiv--2205}, year={2022} }
注意: 此代码仅用于学术目的,不得用于任何可能被视为商业用途的场合。




