1、利用yolov4对安全帽进行识别,对图片数据集进行处理,将图片路径和标签提取出来放到一个txt文本中。然后,将这个txt里的数据和标签按照8:1:1的格式,分为训练集、验证集和测试集。然后进上诉的内容输入的到网络中。再次之前对每一次的输入进行色彩变化、以及一些基础的数据增强算法等。有效的解决了数据的过拟合。最终得到的测试结果达到了83%。
评论