搭建簡單圖片分類的卷積神經網絡(一)-- 訓練模型的圖片數據預處理

一、訓練之前數據的預處理主要包括兩個方面 1、將圖片數據統一格式,以標籤來命名並存到train文件夾中(假設原始圖片按類別存到文件夾中)。 2、對命名好的圖片進行訓練集和測試集的劃分以及圖片數據化。 先對整個項目文件進行說明: 項目文件夾 image文件裏是用來對模型測試的未處理圖片(訓練模型不用)。 logs文件存放訓練好的模型。 prediction文件是image文件中圖片經過模型測試後分類
相關文章
相關標籤/搜索