一种基于改进YOLOv7的自然树种识别方法

摘要

本发明公开了一种基于改进YOLOv7的自然树种识别方法，本发明属于林业树种类识别领域，包括：获取自然树种图像，所述自然树种图像包括：训练图像和测试图像；使用Mosaic‑8数据增强手段对所述训练图像进行数据增强，得到增强训练图像；构建YOLOv7网络，对YOLOv7网络结构进行改进，得到改进YOLOv7模型；其中改进YOLOv7模型包括骨干网络、检测头层网络、注意力机制模块、Rep和Conv，通过检测头层网络输出四层不同尺寸的特征图；基于训练图像训练改进YOLOv7模型，从而得到更好的效果；将测试集图像输入至训练好的改进YOLOv7模型，得到自然树种的识别结果。本发明能够对树木种类进行准确识别。