torch.utils.data和torvision等PyTorch數據處理工具箱

數據處理工具箱概述 Pytorch涉及數據處理（數據裝載，數據預處理，數據增強等）主要工具包及相互關係如圖所示： 圖的左邊是torch.utils.data工具包，主要包括以下4個類： Dataset：是一個抽象類，其他數據集要繼承這個類，並且重寫其中兩個方法（getitem, len）。 DataLoader：定義一個新的迭代器，實現批量（batch）讀取，打亂數據（shuffle）並提供並行加

>>阅读原文<<