Csdn上上传的代码都是完整的,大家也可以下载其他项目练手,丰富自己的简历也不错。
博客演示地址:https://blog.csdn.net/qq_34904125/article/details/124913192
其他小项目完整代码:
https://blog.csdn.net/qq_34904125?type=download
算法部分
本次是口罩目标检测
数据集原图放在009yolov4口罩目标检测识别\dataset\JPEGImages路径下,
对应的xml标签放在009yolov4口罩目标检测识别\dataset\Annotations路径下。
打标签所使用的工具是labelimg工具。
数据集放好后,代码依次运行
python 01create_txt.py
会在本地生成train.txt文本
里面存放的是图片路径和对应的标签。
python 02train.py
会训练得到模型,模型自动保存在weights文件夹下。
python 03predict.py
可以对单张图片进行识别口罩
04pyqt界面.py
展示可视化的qt界面