pytorch創建本身的數據集Dataset

源碼下載，分別在label.py、test.py 將圖片集（下圖class）存到在當前工程目錄下，並按類別分開（下圖的一、二、3）： 爲圖片生成label：python import os path = "class" # 圖片集路徑 classes = [i for i in os.listdir(path)] files = os.listdir(path) train = open("t

>>阅读原文<<