Pytoch經典數據集,蜜蜂和螞蟻數據集(Hymenoptera)
參考資料:Pytoch Hymenoptera
因網站資源託管在Amazon服務器上,所以下載速度可能奇慢無比,本人已將該數據集上傳,下面提供國內下載:
hymenoptera_data數據集,[國內快速下載]
這個數據集包括兩類圖片, bees 和 ants, 這些數據都被處理成了可以使用ImageFolder 來讀取的格式。我們只需要把data_dir
設置成數據的根目錄,然後把model_name
設置成我們想要使用的訓練模型,如:
resnet , alexnet, vgg16等。
Pytoch.Imagefloader的其他參數有
num_classes
表示數據集分類的類別數batch_size
num_epochs
feature_extract
表示我們訓練的時候使用fine tuning還是feature extraction方法。如果feature_extract = False
,整個模型都會被同時更新。如果feature_extract = True
,只有模型的最後一層被更新。
下載鏈接:https://download.csdn.net/download/weixin_46233323/12182815