基于SGR-YOLO模型的安全装备检测方法及系统

本发明提供了一种基于SGR‑YOLO模型的安全装备检测方法及系统，属于安全装备检测技术领域，该方法包括获取安全装备图像，并对安全装备图像进行预处理，将预处理后的安全装备图像划分为训练集、测试集和验证集；利用训练集和验证集对SGR‑YOLO模型进行训练，得到训练后的SGR‑YOLO模型权重；利用评价指标对已训练的SGR‑YOLO模型进行评估，并将测试集输入至已训练的SGR‑YOLO模型中，得到施工人员是否穿戴安全装备的检测结果。本发明解决了在嵌入式设备中落地部署时yolov5算法轻量化差以及检测精度低的问题。