1. 根據pytorch的命令安裝torchvision
2. pytorch的官網提供的例子中使用了cifar-10數據集,需要下載。
通過torchvision下載會很慢,且充滿不確定性,我試了諸多次最后才成功。
一旦成功后,盡管download = true,也不會再下載了。
Files already downloaded and verified
Files already downloaded and verified
還可以直接去官網下載,下載CIFAR-10 python version,然后解壓為cifar-10-batches-py文件夾,
並復制到相對目錄./data下,這個./data為工程的目錄,
例如使用pycharm創建了一個hello的工程:/Users/Aey/PycharmProjects/hello/venv/include/。
3. torchvision.datasets.CIFAR10的參數
root,表示cifar10數據的加載的相對目錄
train,表示是否加載數據庫的訓練集,false的時候加載測試集
download,表示是否自動下載cifar數據集
transform,表示是否需要對數據進行預處理,none為不進行預處理
4. torch.utils.data.DataLoader的參數
batch_size=4 將訓練集的50000張圖片划分成12500份,每份4張圖,用於mini-batch輸入。
shffule=True在表示不同批次的數據遍歷時,打亂順序。
num_workers=2表示使用兩個子進程來加載數據