torchvision的安裝及使用


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表示使用兩個子進程來加載數據


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM