网站首页 网站建设 IT知识 IT知识教程 pytorch训练imagenet分类的方法

pytorch训练imagenet分类的方法

2021-05-22 21:29:28

1、imagenet数据准备:

a、下载数据集

b、提取training data:

mkdir train && mv ILSVRC2012_img_train.tar train/ && cd traintar -xvf ILSVRC2012_img_train.tar && rm -f ILSVRC2012_img_train.tarfind . -name "*.tar" | while read NAME ; do mkdir -p "${NAME%.tar}"; tar -xvf "${NAME}" -C "${NAME%.tar}"; rm -f "${NAME}"; donecd ..

c、提取验证集

mkdir valcp pre_val.sh val/cp ILSVRC2012_img_val.tar val/ && cd val && tar -xvf ILSVRC2012_img_val.tar && rm -f ILSVRC2012_img_val.tarecho "pre_val.sh running"sh pre_val.shrm pre_val.sh

注意:如果最后一步无法成功,打开网页复制内容生成一个pre_val.sh文件后处理

windows编辑的sh文件用下面去问题cat -v old.sh |tr -d "^M" > new.sh

以上这篇pytorch训练imagenet分类的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。 如涉及版权问题,请提交至online#300.cn邮箱联系删除。

以上产品还未完全满足我的所有需求,在下方提交我的专属需求
我的专属需求:
*手机号:
*验证码:
img
咨询报价
现在咨询
img

在线咨询

建站在线咨询

img

微信咨询

扫一扫添加
动力姐姐微信

img
img

TOP