计算机科学 ›› 2024, Vol. 51 ›› Issue (11A): 230900155-7.doi: 10.11896/jsjkx.230900155
陈冬1, 周浩1, 袁国武1, 杨凌宇1, 成秋艳1, 任莹2, 马仪2
CHEN Dong1, ZHOU Hao1, YUAN Guowu1, YANG Lingyu1, CHENG Qiuyan1, REN Ying2, MA Yi2
摘要: 针对输电线路山火中烟雾火灾目标存在尺度多变、背景复杂的特点,会导致检测精度低和误检的问题,文中在YOLOv5的基础上提出了一种融合多尺度特征与位置信息的输电线路山火检测算法。首先,为解决尺度多变的问题,针对SPPF(快速空间金字塔池化层)级联式结构在尺度融合时只关注局部特征信息的问题,提出了一种层级式与CSP(跨阶段局部网络)结构相结合的尺度融合模块SPPCSPC(跨阶段空间金字塔池化层),能有效提取并融合多层次、不同尺度的全局特征信息,提高算法对烟雾火灾目标的检测能力。其次,为解决误检问题,提出了新的颈部网络PCANet(路径位置聚合网络)既融合浅层和深层特征图,又分别从特征图的竖直和水平两个方向将目标的位置信息融入通道中,增强算法对烟雾火灾目标位置信息的关注度,降低复杂背景的干扰。在输电线路烟雾火灾数据集上进行实验,所提算法的mAP50提高了1.6%,mAP50:95提高了1.5%,Recall提高了2.4%,有效提高了烟雾火灾目标检测精度,减少了误检现象发生,能够更好地完成检测任务。
中图分类号:
[1]Y C L,N X,X H K,et al.Overview of Mountain Fire Monitoring and Early Warning for Power Grid Demand [J/OL].Power System Technology.https://doi.org/10.13335/j.1000-3673.pst.2022.2267. [2]DOSOVITSKIY A,BEYER L,KOLESNIKOV A,et al.Animage is worth 16x16 words:Transformers for image recognition at scale[J].arXiv:2010.11929,2020. [3]REN S,HE K,GIRSHICK R,et al.Faster r-cnn:Towards real-time object detection with region proposal networks[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2016,39(6):1137-1149. [4]LIU W,ANGUELOV D,ERHAN D,et al.Ssd:Single shotmultibox detector[C]//Computer Vision-ECCV 2016:14th European Conference,Amsterdam,The Netherlands,Part I 14.Springer International Publishing,2016:21-37. [5]REDMON J,FARHADI A.Yolov3:An incremental improvement[J].arXiv:1804.02767,2018. [6]BOCHKOVSKIY A,WANG C Y,LIAO H Y M.Yolov4:Optimal speed and accuracy of object detection[J].arXiv:2004.10934,2020. [7]CARION N,MASSA F,SYNNAEVE G,et al.End-to-end object detection with transformers[C]//European Conference on Computer Vision.Cham:Springer International Publishing,2020:213-229. [8]JOCHER G,STOKEN A,BORVEC J,et al.YOLOv5[OL].http://doi.org/10.5281/zenodo.4154370(accessed on 16 November 2020). [9]WANG C Y,LIAO H Y M,WU Y H,et al.CSPNet:A newbackbone that can enhance learning capability of CNN[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops.2020:390-391. [10]ZHANG D,ZHANG R Z,LONG Y T,et al.Application Status of Remote Sensing Techniques for Wildfire Monitoring Near Transmission Line Corridor [J].WORLD SCI-TECH R&D,2023,45(2):200-209. [11]LI H D,ZHOU K,ZHANG R Z,et al.Technology Scheme on Wildfire Multidimensional Monitoring of Overhead Transmission Line [J].Rural Electrification,2022(10):10-13. [12]FAN G Y,CHEN X H,CHEN M J,et al.Research and application of satellite remote sensing monitoring fire near transmission lines [J].Mineral Exporation,2021,12(8):1844-1851. [13]LEE Y,SHIM J.False positive decremented research for fireand smoke detection in surveillance camera using spatial and temporal features based on deep learning[J].Electronics,2019,8(10):1167. [14]SUN K,ZHAO Q,WANG X.Using knowledge inference tosuppress the lamp disturbance for fire detection[J].Journal of Safety Science and Resilience,2021,2(3):124-130. [15]SZEGEDY C,IOFFE S,VANHOUCKE V,et al.Inception-v4,inception-resnet and the impact of residual connections on lear-ning[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2017. [16]XUE Z,LIN H,WANG F.A small target forest fire detection model based on YOLOv5 improvement[J].Forests,2022,13(8):1332. [17]LIU S,QI L,QIN H,et al.Path aggregation network for instance segmentation[C]//Proceedings of the IEEE Confe-rence on Computer Vision and Pattern Recognition.2018:8759-8768. [18]HOU Q,ZHOU D,FENG J.Coordinate attention for efficientmobile network design[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition.2021:13713-13722. [19]CAI Z,VASCONCELOS N.Cascade r-cnn:Delving into highquality object detection[C]//Proceedings of the IEEE Confe-rence on Computer Vision and Pattern Recognition.2018:6154-6162. [20]GE Z,LIU S,WANG F,et al.Yolox:Exceeding yolo series in 2021[J].arXiv:2107.08430,2021. [21]LI C,LI L,JIANG H,et al.YOLOv6:A single-stage object detection framework for industrial applications[J].arXiv:2209.02976,2022. [22]WANG C Y,BOCHKOVSKIY A,LIAO H Y M.YOLOv7:Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition.2023:7464-7475. [23]ZHU X,LYU S,WANG X,et al.TPH-YOLOv5:ImprovedYOLOv5 based on transformer prediction head for object detection on drone-captured scenarios[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision.2021:2778-2788. [24]LIN T Y,GOYAL P,GIRSHICK R,et al.Focal loss for dense object detection[C]//Proceedings of the IEEE International Conference on Computer Vision.2017:2980-2988. |
|