Pytorch 基礎學習3: 圖像分類模型訓練

►前言

       上篇文章「Pytorch 基礎學習2_Dataset與DateLoader」,講解Dataset類別可以輕鬆地建立一個自定義的Dataset,並通過DataLoader就能夠在訓練模型時載入資料進行模型訓練,本篇主要講解Kaggle競賽-貓狗影像辨識資料集採用自定義模型並結合DataLoader對於數據進行批量化處理來進行模型訓練,讓大家能夠更了解 PyTorch 在圖像分類辨識的應用 。

►資料預處理






數據處理

















►模型訓練



►模型測試


►結果產出


►小結

透過以上講解,能夠使用Kaggle的貓狗分類數據集建立自定義模型架構,並搭配上一章節Dataset與DataLoader進行數據加載與迭代進行模型訓練,最後將訓練模型結果進行可視化 。下一章節就要講解目標檢測部分,期待下一篇博文吧!

►參考資料    

PyTorch - 練習kaggle - Dogs vs. Cats - 使用自定義的 CNN model

★博文內容均由個人提供,與平台無關,如有違法或侵權,請與網站管理員聯繫。

★文明上網,請理性發言。內容一周內被舉報5次,發文人進小黑屋喔~

評論