飛槳--手勢識別網絡

手勢識別任務主要分爲五個部分: 第一部分是數據的預處理,生成圖像列表,定義訓練集和測試集的reader; 第二部分是網絡的定義,主要包含卷積層、池化層、BN層和全連接層 第三部分是訓練,定義了優化器,損失函數,經過反向傳播進行參數更新; 第四部分是模型檢驗,也就是在測試訓練得到的模型; 第五部分是預測,加載模型測試任意一張圖片。 定義的網絡部分如下圖:
相關文章
相關標籤/搜索