首先介紹下什麼是MTCNN?
論文:Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks (獲取原論文,私信我回復“MTCNN”即可獲得)
MTCNN提出了一種Multi-task的人臉檢測框架,將人臉檢測和人臉特徵點檢測同時進行(我這裡沒有進行人臉特徵點檢測,其實瞭解了你就知道了,多幾行數據代碼就可以了)。論文使用3個CNN級聯的方式。
這個MTCNN的簡介在網上一搜一大把,所以也就不再做太多的簡介和名頭講解了,總結下里面的主要結構點和後續的講解重點。
項目建立說明:
數據集:CelebaA(這個數據集不是很大,大概20G,不過也可以使用3萬多張的進行訓練)
GPU型號:GTX1060 6G(雖然這個網絡不是很大,但是還是GPU運行起來比較快)
框架:Torch
語言:Python
項目流程
MTCNN的主要結構介紹:
![Python MTCNN(人臉檢測)項目附代碼講解(1)-原理與論文介紹](http://p2.ttnews.xyz/loading.gif)
思維導圖的梳理:
![Python MTCNN(人臉檢測)項目附代碼講解(1)-原理與論文介紹](http://p2.ttnews.xyz/loading.gif)
圖一直不清楚,但是這個沒事的,等我講解完,會把pdf上傳到網盤
模型建立code:
訓練優化:
效果展示:
問題總結:
如果你看過我之前的AI建模流程的話,那麼這個流程對於你來說其實並不難,
以上就是後續要講解的所有過程,我就先不做過多的詳細解釋了,大概看下流程,後續會進行數據說明與代碼展現,帶領大家一點一點建立起這個網絡~人臉檢測~
喜歡就關注我哦,持續日更中……
閱讀更多 極客Array 的文章