Pytoch.Imagefloader經典數據集:hymenoptera數據集下載太慢解決方法、數據集介紹和使用

Pytoch經典數據集,蜜蜂和螞蟻數據集(Hymenoptera)

參考資料Pytoch Hymenoptera

因網站資源託管在Amazon服務器上,所以下載速度可能奇慢無比,本人已將該數據集上傳,下面提供國內下載:
hymenoptera_data數據集,[國內快速下載]

這個數據集包括兩類圖片, beesants, 這些數據都被處理成了可以使用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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章