- 博客(4)
- 收藏
- 关注
RSS订阅原创 opencv dnn部署yolov5的口罩检测模型+GUI界面(三)
我训练完自己的口罩检测模型，得到了一个best.pt文件，在detect.py里运行无误后，尝试把模型导出。 yolov5官方代码的训练出来模型的权重是保存在best.pt文件里的。这个.pt后缀的模型文件并不常规，如果想要脱离训练环境部署模型，需要对原始的.pt文件进行导出操作，这里我把模型导出为onnx文件，想要用opencv的dnn模块读取该onnx模型文件进行推理，这样操作之后，整个检测过程就只需要import cv2这一个第三方库了。 yolov...
2021-08-17 16:54:34
4074
14
原创 opencv dnn部署yolov5的口罩检测模型+GUI界面(二)
（二）数据的准备和模型的训练 我做的是口罩检测，数据集是把图像中的人脸框出，再打上是否佩戴口罩的标签，再把数据集按一定的比例划分一下训练集和测试集。 打开yolov5的主目录，点进data里面，把图像和标签分别放在下图的images和labels文件夹下面。 而点进其中的images里，又有这样三个目录，test, train, val。这就是我们的数据集了，val是交叉验证集。我的数据集总共只有1400+张图片，划分了70%训练集，测试集和验证集分别15%。 images的结构如下图 .
2021-08-14 21:05:51
1690
1
原创 opencv dnn部署yolov5的口罩检测模型+GUI界面(一)
记录一下大学里自己完成的第一个小课题。 任务是人脸口罩检测，使用yolov5训练目标检测模型，在训练环境下测试无误后，再导出为.onnx文件，使用opencv的dnn模块进行模型的部署，再简单写一个图形化界面做展示。模型训练是python ...
2021-08-13 13:48:54
2546
原创 深度学习的Hello world 基于MNIST库的手写数字识别
实验基于MNIST库的手写数字识别 目录 实验基于MNIST库的手写数字识别 1 实验思路 2 MNIST数据集的加载和归一化 3 CNN的搭建 4 训练和测试及其可视化 5 实验中遇到的问题 6 全部代码 MNIST 数据集来自美国国家标准与技术研究所, 训练集由来自 250 个不同人手写的数字构成, 其中 50% 是高中学生, 50% 来自人口普查局MNIST数据集是机器学习领域中非常经典的一个数据集，由60000个训练样本和10000个测试样本组成，每个样本都是一...
2021-05-17 23:08:58
438
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人