【使用AutoDL运行开放步态识别框架OpenGait】_casia-b

最近在使用AutoDL云服务器跑步态识别的代码，用的是开放步态识别框架OpenGait，记录一下自己跑代码的步骤和遇到的问题以及解决方式。

一、上传OpenGait项目到AutoDL

AutoDL网址：https://www.autodl.com/home
OpenGait网址：https://github.com/ShiqiYu/OpenGait/

二、CASIA-B数据预处理

1、下载CASIA-B数据集
CASIA-B下载网址：http://www.cbsr.ia.ac.cn/GaitDatasetB-silh.zip
2、上传CASIA-B数据集到AutoDL
3、解压数据集→进入解压后的目录→再次解压

unzip CASIA-B
cd CASIA-B
sudo ls *.tar.gz | xargs -n1 tar xzvf
1
2
3

4、数据预处理，代码中的预处理主要包括①裁剪②转为pkl文件

python datasets/pretreatment.py --input_path /root/CASIA-B --output_path /root/CASIA-B-pkl
1

三、运行Gaitbase

# 4块卡跑
# 训练
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 opengait/main.py --cfgs ./configs/gaitbase/gaitbase_da_casiab.yaml --phase train

# 测试
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 opengait/main.py --cfgs ./configs/gaitbase/gaitbase_da_casiab.yaml --phase test
1
2
3
4
5
6

CUDA_VISIBLE_DEVICES=0：哪一块显卡
nproc_per_node=1： GPU数量
根据服务器GPU数量进行调整

四、运行结果

五、问题及解决

RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
模型的训练的batch-size训练过大了，调整更小，就可以了

总结

欢迎使用OpenGait研究步态识别的伙伴进行交流