學(xué)習(xí)深度學(xué)習(xí)已有一段時(shí)間了,總想著拿它做點(diǎn)什么,今天終于完成了一個(gè)基于caffe的人臉檢測(cè),這篇博文將告訴你怎樣通過(guò)caffe一步步實(shí)現(xiàn)人臉檢測(cè)。本文主要參考唐宇迪老師的教程,在這里感謝老師的辛勤付出。
傳統(tǒng)機(jī)器學(xué)習(xí)方法實(shí)現(xiàn)人臉檢測(cè):
人臉檢測(cè)在opencv中已經(jīng)幫我們實(shí)現(xiàn)了,我們要把它玩起來(lái)很簡(jiǎn)單,只需要簡(jiǎn)簡(jiǎn)單單的幾行代碼其實(shí)就可以搞定。(haarcascade_frontalface_alt.xml這個(gè)文件在opencv的安裝目錄下能找到,筆者的路徑是:E:\opencv2.4.10\opencv\sources\data\haarcascades,大家可根據(jù)自己的安裝路徑找到)
face_detect.cpp
運(yùn)行結(jié)果:
caffe實(shí)現(xiàn)人臉檢測(cè):
我是在ubuntu16.04環(huán)境下完成的實(shí)驗(yàn),渣渣筆記本有不起GPU跑訓(xùn)練,所有實(shí)驗(yàn)也是基于CPU的。要想把人臉檢測(cè)玩起來(lái),首先你得保證你的ubuntu已經(jīng)安裝了opencv和caffe,初次配這兩個(gè)環(huán)境初學(xué)者往往會(huì)弄到吐血,而且還是吐老血,我自己已經(jīng)記不清到底花了多久才把它們搞定(估計(jì)是我太慫,也許你很快就能弄好喲,加油)。這里給兩個(gè)參考鏈接,opencv在ubuntu下的配置和測(cè)試:http://blog.csdn.net/a1429331875/article/details/31539129;ubuntu16.04上caffe的配置與安裝(CPU ONLY):http://blog.csdn.net/u010402483/article/details/51506616;以上兩個(gè)鏈接僅供參考,配置過(guò)程出了問(wèn)題大家就多去網(wǎng)上搜解決方案吧,總會(huì)有人也遇到過(guò)和你一樣的問(wèn)題。配置好以后大家可以先跑跑MNIST手寫(xiě)字體識(shí)別這個(gè)案例吧,這個(gè)案例算是給自己的一個(gè)安慰。 到這里就已經(jīng)默認(rèn)大家環(huán)境已經(jīng)配置好了。
第一步:(這樣寫(xiě)感覺(jué)很蠢,但還是寫(xiě)得盡量詳細(xì)吧)在桌面或者你喜歡的路徑下建一個(gè)文件夾,這個(gè)文件夾將用來(lái)存放我們實(shí)驗(yàn)中用到的所有東西。我自己是在桌面建了一個(gè)文件夾,取名:faceDetect
第二步:獲取人臉和非人臉圖片當(dāng)作訓(xùn)練集和驗(yàn)證集。首先我們一定要有樣本,在本實(shí)驗(yàn)中我們的樣本是一個(gè)二分類(lèi)的樣本,大家可以自行去網(wǎng)上找數(shù)據(jù)集,當(dāng)然也可以給我發(fā)郵件(likai_uestc@163.com),我這里有數(shù)據(jù)集。數(shù)據(jù)集我們分訓(xùn)練集(trainData20000_20000)和驗(yàn)證集(testData1600_1600),trainData20000_20000文件夾和testData1600_1600文件夾我們把它們兩個(gè)都放在faceDetect文件夾下,trainData2000