pytorch一步一步在VGG16上訓練自己的數據集

準備數據集及加載，ImageFolder 在很多機器學習或者深度學習的任務中，往往我們要提供自己的圖片。也就是說我們的數據集不是預先處理好的，像mnist，cifar10等它已經給你處理好了，更多的是原始的圖片。比如我們以貓狗分類爲例。在data文件下，有兩個分別爲train和val的文件夾。然後train下是cat和dog兩個文件夾，裏面存的是自己的圖片數據，val文件夾同train。這樣我們的

>>阅读原文<<