底下代碼將載入所有圖片並訓練是否有載口罩,此範例使用 vgg19 模型進行判定,然後儲存模型及權重。經本人測試,VGG19 比 VGG16 更精準。圖片資料請到 https://github.com/prajnasb/observations 下載。將圖檔解開後,把observations-master\experiements\data copy 到專案目錄下
執行此範例前,請先安裝如下套件
pip install tensorflow opencv-python matplotlib
模型訓練.py
底下是口罩模型訓練的完整代碼

