opencv dnn部署yolov5的口罩检测模型+GUI界面(二)

（二）数据的准备和模型的训练

我做的是口罩检测，数据集是把图像中的人脸框出，再打上是否佩戴口罩的标签，再把数据集按一定的比例划分一下训练集和测试集。

打开yolov5的主目录，点进data里面，把图像和标签分别放在下图的images和labels文件夹下面。

而点进其中的images里，又有这样三个目录，test, train, val。这就是我们的数据集了，val是交叉验证集。我的数据集总共只有1400+张图片，划分了70%训练集，测试集和验证集分别15%。

images的结构如下图

同样，labels里的标签也要和images里的图片一一对应。