TensorFlow搭建卷積網絡識別交通標誌

目標是修改LeNet用於識別德國交通標誌,共43個類別。LeNet有兩個卷積層、三個全鏈接層,我增長了一層卷積,修改了kernel大小。另外,在全鏈接層中加入了dropou用於防止過擬合。這裏記錄下搭建網絡的過程,總結使用TensorFlow搭建並訓練卷積網絡的方法。python 1. 數據集的載入、augment與normalization 載入數據集 """ 載入數據集 """ # 下載好的數
相關文章
相關標籤/搜索