【深度之眼】Pytorch框架班第五期-圖像分類一瞥

模型是如何進行圖像分類的？ 圖像分類的Inference（推理） 步驟： 1、獲取數據與標籤 2、選擇模型，損失函數，優化器 3、寫訓練代碼 4、寫Inference代碼 Inference代碼基本步驟： 1、獲取數據與模型 2、數據變換，如RGB -> 4D-Tensor 3、前向傳播 4、輸出保存預測結果 Inference階段注意事項： 1、確保model處於eval狀態而非training

>>阅读原文<<