deep-high-resolution-net.pytorch训练AnimalKingdom数据集

还未完成,挖个坑先

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!
,提示缺少预训练模型此时需要到官网去下载对应的模型

https://onedrive.live.com/?authkey=%21AEwfaSueYurmSRA&id=56B9F9C97F261712%2111773&cid=56B9F9C97F261712

下载完成后放入报错所提示的文件夹,没有的话自己新建

 运行后又报错,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还是无解,等着换显卡或者放到平台上跑吧

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值