还未完成,挖个坑先
1、首先下载AnimalKindom数据集,安装所需环境
运行pose_estimation\code\hrnet\tools下的train.py,报错train.py: error: the following arguments are required: --cfg,需要我们指定配置文件,我选pose_estimation\code\hrnet\experiments\mpii\hrnet下的w32_256x256_adam_lr1e-3.yaml
在IDE中如图填入配置文件路径保存,同时将工作目录设置为hrnet(写入绝对路径,图上有误)
运行后又报错ValueError: models/pytorch/imagenet/hrnet_w32-36af842e.pth is not exist!
,提示缺少预训练模型此时需要到官网去下载对应的模型
下载完成后放入报错所提示的文件夹,没有的话自己新建
运行后又报错,AssertionError: Invalid device id
经过排查是我的gpu只有一个,而配置文件里是一个元组有4个(0,1,2,3),将它修改为1个(0,)
运行又报错AttributeError: module 'dataset' has no attribute 'mpii',打开配置文件修改
DATASET:ak
ROOT:为存放训练和测试json的路径
添加IMAGE_ROOT:为你的训练图片地址
运行又报错,爆显存了,batch_size修改为1还是无解,等着换显卡或者放到平台上跑吧