改进的YOLOv5轻量级草地烟雾检测算法

Jiaxin Su, Zhiqiang Liu, Xu Zhang, Wenjing Li, Mixue Zhu
{"title":"改进的YOLOv5轻量级草地烟雾检测算法","authors":"Jiaxin Su, Zhiqiang Liu, Xu Zhang, Wenjing Li, Mixue Zhu","doi":"10.1145/3573428.3573724","DOIUrl":null,"url":null,"abstract":"To address the problems of low detection performance and large memory consumption of traditional smoke and fire detection methods in complex scenes such as grasslands. Based on the YOLOv5 model, a YOLOv5-GDE optimization model is proposed. The C3 module in YOLOv5 is replaced by GhostC3 with a smaller number of parameters, and some standard convolution blocks are replaced by depth-separable convolutions to make the model more lightweight. Finally, to solve the problem of unstable target regression frame, the EIoU loss function is introduced, which effectively improves the convergence speed and detection accuracy of the model. Experimental results on the homemade grassland smoke dataset show that the optimized model reduces the number of parameters and computational effort by 65.4% and 65.8%, respectively, compared with the original model, and the model size is only 36.8% of the original model, which is more suitable for smoke target detection in grassland scenes and more suitable for deployment in embedded devices with limited computational power, under the premise of ensuring detection accuracy.","PeriodicalId":314698,"journal":{"name":"Proceedings of the 2022 6th International Conference on Electronic Information Technology and Computer Engineering","volume":"1 1","pages":"0"},"PeriodicalIF":0.0000,"publicationDate":"2022-10-21","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":"{\"title\":\"Improved YOLOv5 lightweight grassland smoke detection algorithm\",\"authors\":\"Jiaxin Su, Zhiqiang Liu, Xu Zhang, Wenjing Li, Mixue Zhu\",\"doi\":\"10.1145/3573428.3573724\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"To address the problems of low detection performance and large memory consumption of traditional smoke and fire detection methods in complex scenes such as grasslands. Based on the YOLOv5 model, a YOLOv5-GDE optimization model is proposed. The C3 module in YOLOv5 is replaced by GhostC3 with a smaller number of parameters, and some standard convolution blocks are replaced by depth-separable convolutions to make the model more lightweight. Finally, to solve the problem of unstable target regression frame, the EIoU loss function is introduced, which effectively improves the convergence speed and detection accuracy of the model. Experimental results on the homemade grassland smoke dataset show that the optimized model reduces the number of parameters and computational effort by 65.4% and 65.8%, respectively, compared with the original model, and the model size is only 36.8% of the original model, which is more suitable for smoke target detection in grassland scenes and more suitable for deployment in embedded devices with limited computational power, under the premise of ensuring detection accuracy.\",\"PeriodicalId\":314698,\"journal\":{\"name\":\"Proceedings of the 2022 6th International Conference on Electronic Information Technology and Computer Engineering\",\"volume\":\"1 1\",\"pages\":\"0\"},\"PeriodicalIF\":0.0000,\"publicationDate\":\"2022-10-21\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"0\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"Proceedings of the 2022 6th International Conference on Electronic Information Technology and Computer Engineering\",\"FirstCategoryId\":\"1085\",\"ListUrlMain\":\"https://doi.org/10.1145/3573428.3573724\",\"RegionNum\":0,\"RegionCategory\":null,\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"\",\"JCRName\":\"\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"Proceedings of the 2022 6th International Conference on Electronic Information Technology and Computer Engineering","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1145/3573428.3573724","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"","JCRName":"","Score":null,"Total":0}
引用次数: 0

摘要

针对草原等复杂场景下传统的烟、火探测方法检测性能低、内存消耗大的问题。在YOLOv5模型的基础上,提出了YOLOv5- gde优化模型。将YOLOv5中的C3模块替换为参数数量较少的GhostC3,并将一些标准卷积块替换为深度可分卷积,使模型更加轻量化。最后,针对目标回归框架不稳定的问题,引入EIoU损失函数,有效提高了模型的收敛速度和检测精度。在自制草原烟雾数据集上的实验结果表明,优化后的模型与原始模型相比,参数数量和计算量分别减少了65.4%和65.8%,模型尺寸仅为原始模型的36.8%,在保证检测精度的前提下,更适合草原场景中的烟雾目标检测,更适合部署在计算能力有限的嵌入式设备中。
本文章由计算机程序翻译,如有差异,请以英文原文为准。
Improved YOLOv5 lightweight grassland smoke detection algorithm
To address the problems of low detection performance and large memory consumption of traditional smoke and fire detection methods in complex scenes such as grasslands. Based on the YOLOv5 model, a YOLOv5-GDE optimization model is proposed. The C3 module in YOLOv5 is replaced by GhostC3 with a smaller number of parameters, and some standard convolution blocks are replaced by depth-separable convolutions to make the model more lightweight. Finally, to solve the problem of unstable target regression frame, the EIoU loss function is introduced, which effectively improves the convergence speed and detection accuracy of the model. Experimental results on the homemade grassland smoke dataset show that the optimized model reduces the number of parameters and computational effort by 65.4% and 65.8%, respectively, compared with the original model, and the model size is only 36.8% of the original model, which is more suitable for smoke target detection in grassland scenes and more suitable for deployment in embedded devices with limited computational power, under the premise of ensuring detection accuracy.
求助全文
通过发布文献求助,成功后即可免费获取论文全文。 去求助
来源期刊
自引率
0.00%
发文量
0
×
引用
GB/T 7714-2015
复制
MLA
复制
APA
复制
导出至
BibTeX EndNote RefMan NoteFirst NoteExpress
×
提示
您的信息不完整,为了账户安全,请先补充。
现在去补充
×
提示
您因"违规操作"
具体请查看互助需知
我知道了
×
提示
确定
请完成安全验证×
copy
已复制链接
快去分享给好友吧!
我知道了
右上角分享
点击右上角分享
0
联系我们:info@booksci.cn Book学术提供免费学术资源搜索服务,方便国内外学者检索中英文文献。致力于提供最便捷和优质的服务体验。 Copyright © 2023 布克学术 All rights reserved.
京ICP备2023020795号-1
ghs 京公网安备 11010802042870号
Book学术文献互助
Book学术文献互助群
群 号:481959085
Book学术官方微信