[发明专利]一种基于改进YOLOv5的红树林单木目标检测方法

权利要求书

1.一种基于改进YOLOv5的红树林单木目标检测方法，其特征在于含有以下步骤：

利用开源软件LabelImg在选取的无人机影像上依次标注目标树，构建红树林单木数据集；

选择YOLOv5为基础目标检测模型，依据目标木分布密集且尺寸较小的特点对其进行优化和改进，使用有效通道注意力机制Efficient Channel Attention对CSPDarknet53骨干网络进行改进，避免降维和具有跨通道交互作用的同时，增强特征表达能力，并在SPP模块引入SoftPool池化操作，保留更多细节特征信息，提高自动化目标检测精度。

2.根据权利要求1所述的一种基于改进YOLOv5的红树林单木目标检测方法，其特征在于包括步骤如下：

步骤1、由于幼小的红树林单木特殊的地理位置以及分布密集且目标较小，借助无人机独有的时间灵活、高空间分辨率优势去获得数据以构建所需要的数据集；

步骤2、在YOLOv5的CSPDarknet53特征提取网络中引入有效通道注意力机制(Efficient Channel Attention)模块，对YOLOv5网络进行改进，改进后模型重新命名为YOLOv5-ECA；

步骤3、在步骤2的基础上进一步改进，在YOLOv5网络SPP模块中引入SoftPool改进池化操作，保留更多细节特征信息；

步骤4、将得到的无人机红树林单木数据集图像输入到Yolov5特征提取网络进行特征提取，得到不同尺度的特征图；

步骤5、将步骤4中得到的特征图进行分类和回归，将回归结果进行特征重建操作，得到更加精细的特征图，在此基础上再次进行分类和回归操作，计算损失；

步骤6、完成模型的训练后，借助已划分好的数据集对测试集进行测试以实现对幼小红树林单木的目标检测，并对模型的检测效果进行评价。

3.根据权利要求2所述的一种基于改进YOLOv5的红树林单木目标检测方法，其特征在于步骤1中为保证图像中各个检测目标的完整，使用LabelImg开源软件对无人机图像中红树林单木幼苗依次进行标注，标注内容为新造红树林矩形包围框坐标，存储为XML文本文件用于YOLOv5模型的训练和测试，YOLOv5的结构包括输入端Input、基准网络Backbone、特征融合部分Neck、和检测头Head部分，YOLOv5的输入端包含了训练数据集图像的预处理阶段，在输入端沿用YOLOv4的Mosaic数据增强，用来提升模型的训练速度和网络精度，同时新加了自适应锚框计算程序，在骨干网络部分，YOLOv5增加Focus结构，用来提取通用的特征，并依此构造了两种CSP结构，Focus结构对图片进行切片操作，在骨干网络和头检测中间的特征融合部分，Yolov5添加FPN+PAN的组合结构，同时编写相应的分割程序将图片裁剪成所需大小并在裁剪后的图像上将标注目标框贴合在目标树上，核查有无漏标注的目标物，以便补充完整。

4.根据权利要求2所述的一种基于改进YOLOv5的红树林单木目标检测方法，其特征在于步骤2中有效通道注意力机制(Efficient Channel Attention)模块通过大小为k的快速一维卷积产生通道注意，其中核的大小完全由通道维数相关函数自适应来确定，在维度保持不变的条件下输入特征图像χ，所有通道经过全局平均池化之后，ECA模块利用一个可以权重共享的一维卷积来学习特征，并在学习特征的时候涉及到每一个通道与之k个近邻来捕获跨通道交互，k表示快速一维卷积的内核大小，而自适应k取值的确定，是通过跨通道信息交互作用的遮盖区域与通道维数C的正比关系得到，计算如公式(1)所示：

式中：γ＝2，b＝1，|*|_ood表示最邻近奇数，C为通道维数。