最近武漢肺癌很嚴重，公司同事聊到說可不可能做一個口罩偵測的模型，就做了這個Side Project。
模型用 tf.keras 內建的預訓練模型 MobileNetV2 來訓練，資料集從網路上找有/無口罩各400張的臉而成。
準備好資料集後，執行 train.py 可進行訓練。 將預訓練模型放到 exported 底下，執行 webcam_demo.py 可開啟鏡頭測試。
本專案仍非常簡陋，在遇到用手遮擋嘴巴的狀況下，人會辨識為有戴口罩，若要加強此部分，請在訓練資料集中多加入特殊樣本。
最近武漢肺癌很嚴重，公司同事聊到說可不可能做一個口罩偵測的模型，就做了這個Side Project。
模型用 tf.keras 內建的預訓練模型 MobileNetV2 來訓練，資料集從網路上找有/無口罩各400張的臉而成。
準備好資料集後，執行 train.py 可進行訓練。 將預訓練模型放到 exported 底下，執行 webcam_demo.py 可開啟鏡頭測試。
本專案仍非常簡陋，在遇到用手遮擋嘴巴的狀況下，人會辨識為有戴口罩，若要加強此部分，請在訓練資料集中多加入特殊樣本。