1. 程式人生 > >基於深度學習的性別分類

基於深度學習的性別分類

所謂性別分類指的是給定一張圖片,輸出其中人臉的性別(男性或者女性),由於準確率比之前有大幅提升,基於深度學習的方法目前佔據了主流,CVPR2015上曾發表了一篇基於AlexNet的論文,詳細分析參見基於 CNN的年齡和性別檢測

其存在的問題如下:

1是結構偏舊,畢竟15年很多模型都還沒有出來

2是複雜度高,不適合應用到嵌入式端

3是資料問題,由於用於訓練的圖片僅含有正臉的資料,因此它對於正臉的分類效果還算差強人意,稍微有點傾斜或者模糊基本就慘不忍睹

作者給出的程式碼是基於caffe的,大約4個小時就可以訓練完成,精度在95%左右

當把其移植到計算棒的時候Profile下時間就會發現conv2層佔了幾乎一半的執行時間,因此有必要對其進行優化