当前位置:   article > 正文

opencv dnn部署yolov5的口罩检测模型+GUI界面(一)_yolov5和gui

yolov5和gui

记录一下大学里自己完成的第一个小课题,参考了许多大神的经验才磕磕绊绊做出来。

(一)代码的下载和测试

任务是人脸口罩检测,使用yolov5训练目标检测模型,在训练环境下测试无误后,再导出为.onnx文件,使用opencv的dnn模块进行模型的部署,再简单写一个图形化界面做展示。

dnn读取模型的部分有python和C++两个版本。

模型训练直接使用的u神开源在github上的yolov5。

github传送门

下载好之后完整目录如下图(makeData.py是我用来准备自己的数据的,不用管)

 接下来需要配置环境,按照requirements.txt里的需求下载依赖包。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/274665
推荐阅读
相关标签
  

闽ICP备14008679号