首先根據(jù)上個視頻搭建yolov5環(huán)境

將口罩檢測的數(shù)據(jù)集copy到目錄

制作口罩數(shù)據(jù)的yaml文件，命名mask.yaml

修改模型的yaml文件，命名mask.yaml

訓練數(shù)據(jù)，配置對應(yīng)路徑

開始訓練，我在cpu上迭代100次用時7hours

訓練完之后，有對應(yīng)的文件，找到best.pt

接下來就是進行檢測的時候了

檢測結(jié)果

結(jié)果還是不太理想，這里我迭代了100次，建議迭代300次

到此，一名小白的口罩檢測基本完成

接下來訓練更換的best.pt，讓自己的模型更好更快更準確?。?！

謹以此文記錄小白的學習！