1.下载数据集DukeMTMC并转换成遮挡数据集occulded-DukeMTMC,用指令转换python convert_duke_to_occduke.py /path/to/DukeMTMC-reID.zip
2.下载预训练模型pose_hrnet_w48_256x192.pth,而网址打不开,所以从CSDN下载,直接搜pose_hrnet_w48_256x192.pth放到./core/models/model_keypoints/pose_hrnet_w48_256x192.pth
3.训练python main.py --mode train \--duke_path path/to/occluded/duke \--output_path ./results ,我在main.py中修改的路径,duke_path设为数据集绝对路径,output_path设为./results
4.出现问题:找不到文件pose_hrnet_w48_256x192.pth,在./core/models/model_keypoints/config/default.py的131行,改为绝对路径
5.又出现runtimeerror,改batch size,代码中没说batch size,但论文中有说batch size是64个人每人4张图,于