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分類的方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持武林站長站。