赞
踩
记录一下大学里自己完成的第一个小课题,参考了许多大神的经验才磕磕绊绊做出来。
任务是人脸口罩检测,使用yolov5训练目标检测模型,在训练环境下测试无误后,再导出为.onnx文件,使用opencv的dnn模块进行模型的部署,再简单写一个图形化界面做展示。
dnn读取模型的部分有python和C++两个版本。
模型训练直接使用的u神开源在github上的yolov5。
下载好之后完整目录如下图(makeData.py是我用来准备自己的数据的,不用管)
接下来需要配置环境,按照requirements.txt里的需求下载依赖包。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。