防灾减灾工程学报2023，Vol.43Issue(6)：1215-1223,9.DOI:10.13409/j.cnki.jdpme.20230105002
融合YOLOX和ASFF的高原山地灾害检测模型
A Plateau Mountain Disaster Detection Model by Integrating YOLOX and ASFF
何明杰 1刘德方 1张猛 1李高会1
作者信息
- 1. 中国电建集团华东勘测设计研究院有限公司,浙江 杭州 310000
- 折叠
摘要
Abstract
YOLOX is the world's earliest first-order anchor-free frame target detection model,which surpasses traditional anchor frame models such as YOLO-V(3-5)and SSD,and greatly improves the accuracy of mountain disaster detection and recognition.However,the model suffers from feature in-consistency between features at different scales,and the quality of the fused feature map needs to be improved.This paper presents the mountain disaster-stricken areas on the Tibetan plateau as the ex-perimental site,based on the basis of the established Tibetan plateau mountain disaster dataset,intro-duces the Adaptively Spatial Feature Fusion(ASFF)attention mechanism and backbone network size adjustment mechanism to design an optimized system with different applicability.The new plateau mountain disaster detection model(PMDDM)improves the detection accuracy of plateau mountain hazards and enhances the convenience of plateau mountain hazard detection.Meanwhile,the PMD-DM model was also compared with traditional YOLOX and other models with different attention mechanisms and different object detection to verify it superiority.The detection performance and visu-alization results of different scale models are also comparatively analyzed and studied in this paper.The results show that the ASFF attention mechanism can effectively solve the problem of feature in-consistency among different scale features in the traditional YOLOX model;the ASFF attention mechanism improves the model detection performance much better than the attention mechanisms of SE,CBAM,ECA,GAM and Coord;the detection accuracy of the PMDDM model for mountain hazards is better than that of Faster-RCNN,SSD and YOLO-V3 models;the PMDDM model can meet requirements of hardware configuration,detection speed and accuracy for different working sce-narios;the larger the scale of PMDDM model,the higher the accuracy rate of identifying targets.关键词
高原山地灾害/目标检测/注意力机制/YOLOX/ASFFKey words
plateau mountain hazards/object detection/attention mechanism/YOLOX/ASFF分类
建筑与水利引用本文复制引用
何明杰,刘德方,张猛,李高会..融合YOLOX和ASFF的高原山地灾害检测模型[J].防灾减灾工程学报,2023,43(6):1215-1223,9.
