用PyTorch創建一個圖像分類器?So easy!(Part 2)

在第一部分中,我們知道了爲什麼以及如何加載預先訓練好的神經網絡,我們可以用自己的分類器代替已有神經網絡的分類器。那麼,在這篇文章中,我們將學習如何訓練分類器。 訓練分類器 首先,我們需要爲分類器提供待分類的圖像。本文使用ImageFolder加載圖像,預訓練神經網絡的輸入有特定的格式,因此,我們需要用一些變換來調整圖像的大小,即在將圖像輸入到神經網絡之前,對其進行裁剪和標準化處理。 具體來說,將圖
相關文章
相關標籤/搜索