pytorch 加載自己的數據集

pytorch 加載自己的數據集 pytorch 加載自己的數據集，需要寫一個繼承自torch.utils.data中Dataset類，並修改其中的__init__方法、__getitem__方法、__len__方法。默認加載的都是圖片，__init__的目的是得到一個包含數據和標籤的list，每個元素能找到圖片位置和其對應標籤。然後用__getitem__方法得到每個元素的圖像像素矩陣和標籤，返

>>阅读原文<<