demo仓库和视频教程找012期:
到此一游7758258的个人空间_哔哩哔哩_bilibili
![](https://i-blog.csdnimg.cn/blog_migrate/a215c5bfbd7e8f6c9bdf37b81fb98e61.jpeg)
![](https://i-blog.csdnimg.cn/blog_migrate/803ae77f0df99a1060bb40850b36b121.png)
代码文件界面:
![](https://i-blog.csdnimg.cn/blog_migrate/2cfb22ed216c7ef1582d5b5aba6358cb.png)
![](https://i-blog.csdnimg.cn/blog_migrate/53a3ea3cc73952adb35072ec3d953106.png)
其中dataset文件夹下放的是口罩数据集,有图片和对应的标签文件,标签文件记录的是口罩的位置信息。
通过运行01create_txt.py可以将图片路径保存在txt文本里,
在运行02train.py训练图片保存模型到weights文件下。
通过运行03predict.py可以实现单个图片中口罩是否佩戴和是否规范的检测。
04pyqt界面.py是打开一个可视化的ui界面,提供交互使用。