深度學習優質實踐案例：用 Keras 實現人羣中的口罩檢測

Keras 是一個由 Python 編寫的開源人工神經網絡庫，可以作爲 Tensorflow、Microsoft-CNTK 和 Theano 的高階應用程序接口，進行深度學習模型的設計、調試評估、應用和可視化。

Gitee 上這位優秀的開發者也是一名深度學習愛好者，爲了做一次目標檢測學習了 RCNN，FastRCNN，FasterRCNN 的理論，隨後便誕生了這個項目，我們一起來看看他是怎麼做的吧。

 

項目名稱：keras 口罩檢測

項目作者：唐振超

開源許可協議：GPL-3.0

 

項目地址：https://gitee.com/tang_zhen_chao/keraskouzhaojiance

 

 

介紹

使用 keras 搭建 fasterRCNN，在 VOC 格式的口罩數據集上訓練，達到了檢測人羣中有無戴口罩的目的。

 

軟件架構

*可前往項目主頁查看

 

安裝教程

1. 需要numpy,matplotlib,scikit-learn,Pillow,tensorflow1.x,keras
2. pip install package或者conda install package
3. 如果要訓練，需要使用VOC格式數據集，需要安裝labelimg

 

使用說明

1. ./model_data/logs下必須有模型權重文件，由於權重較大，所以未上傳
2. ./theory下是做這個東西的流程與方便理解代碼的原理說明
3. ./net下是fasterRCNN的模型構建
4. run.py用於直接運行看結果，voctrain.py用於訓練自己的VOC數據集

 

結果顯示

 

點擊鏈接前往項目主頁看看項目的詳細信息：https://gitee.com/tang_zhen_chao/keraskouzhaojiance