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}
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.