基于YOLOv3-HA的滑坡区房屋识别

HOUSE IDENTIFICATION IN LANDSLIDE AREA BASED ON YOLOV3-HA

  • 摘要: 针对YOLOv3模型体积大、效率低的问题,提出一种基于YOLOv3改进的滑坡区房屋识别方法YOLOv3-HA。使用HetConv替换常规卷积核,引入CBAM模块和金字塔池化结构改善模型性能,再使用更精确的EIoU作为边框回归损失。基于滑坡房屋数据集的实验结果表明,该模型体积与FLOPs相较于原始模型减小约70%,检测速度提升20%,检测精度提高4.27百分点,验证了该轻量化算法的有效性。

     

    Abstract: Aimed at the problem of large size and low efficiency of the YOLOv3 model, YOLOv3-HA, an improved method for identifying houses in landslide areas based on YOLOv3, is proposed. HetConv was used to replace the conventional convolution kernel, the CBAM module and the pyramid pooling structure were introduced to improve the performance of the model, and a more accurate EIoU was used as the frame regression loss. The experimental results on the landslide house dataset show that compared with the original model, the size of the model and FLOPs are reduced by about 70%, detection efficiency are improved by 20%, and the detection accuracy is increased by 4.27 percentage points, which verifies the effectiveness of the lightweight algorithm.

     

/

返回文章
返回