面向危险施工机械智能识别的改进YOLOv7算法
作者:柯炳明; 叶立威; 廖日旭; 蔡文辉; 戚远航
加工时间:2025-01-16
信息来源:电子设计工程
关键词:YOLO;目标检测;危险机械;输电线路;智能识别
摘 要:为了实现输电线路上危险施工机械(挖掘机、推土机和打桩机)的智能识别,该文提出了一种改进的YOLOv7算法。通过爬虫获取复杂条件下的危险施工机械图像并建立数据集,结合Mixup和Mosaic等多种图像增强方法预处理数据集。采用K-means++算法改进YOLOv7,提高算法的识别效率。实验结果证明,K-means++算法提高了YOLOv7的模型收敛速度;所提出算法能有效进行危险施工机械的识别,识别效果优于YOLOv7,准确率提高了3.70%。
内 容:原文可通过湖北省科技资源共享服务平台(https://www.hbsts.org.cn/)获取