用自己訓練好的caffemodel來對自己的圖片做分類

首先是deploy.prototxt文件的生成,deploy.prototxt和train_val.prototxt文件類似,只是頭尾有些區別而已。沒有了第一層的數據層,也沒有最後的accuracy層(用於反向傳播),但最後多了一個Softmax概率層(Softmax直接計算前向的概率),所以deploy文件沒有了反向傳播部分。 一般不推薦用代碼來生成,熟悉train_val.prototxt的可
相關文章
相關標籤/搜索