摘要
本发明公开了一种基于改进YOLOv8模型的轻量化安全帽检测方法和系统,包括以下步骤:步骤S1,收集安全帽数据集,按照8:2的比例划分为训练集和测试集,并对图片进行标注;步骤S2,构建基于改进YOLOv8模型的轻量化网络模型;步骤S3,使用训练集训练所述基于改进YOLOv8模型的轻量化网络模型,得到最优权重文件;步骤S4,将测试集输入步骤S3得到的最优权重文件进行测试分析。采用本方法模型实现了轻量化,降低了计算量,具有广泛的应用前景和重要的实际意义。
技术关键词
安全帽检测方法
佩戴安全帽
数据收集模块
模型训练模块
检测头
网络
训练集
分析模块
超参数
图片
错误检测
样本
索引
标识符
通道
图像
格式