1. 程式人生 > >MobileFaecNet實現58個人臉關鍵點檢測(附訓練原始碼)

MobileFaecNet實現58個人臉關鍵點檢測(附訓練原始碼)

MobileFaecNet實現58個人臉關鍵點檢測

1.0 caffe修改

實現新網路時,新設計的layer需要自己完成程式碼並新增到caffe中,同時由於人臉關鍵點是迴歸任務,因此為多標籤資料,需要修改資料層image_data_layer從而可以支援多標籤輸入。(但是caffe之前都不支援多標籤輸入嗎?)

image_data_layer.cpp最後幾行原始程式碼:
在這裡插入圖片描述
image_data_layer.cpp最後幾行修改後程式碼:
在這裡插入圖片描述
在caffe_path/include/caffe/layers路徑下加入新增模組程式碼,其中需要注意的是,一般需要新增三個檔案,分別是: .cpp .hpp .cu三個檔案為字尾的檔案,

hpp和cpp檔案的具體說明見此連結,cu檔案是因為要用到cuda加速所配套的檔案。(待補充)

參考連結

連結1
程式碼1
連結2
程式碼2