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

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

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

項目名稱:keras 口罩檢測架構

項目做者:唐振超學習

開源許可協議:GPL-3.0spa

項目地址: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

使用說明ip

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

結果顯示

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

相關文章
相關標籤/搜索