摘要
本发明公开了一种基于YOLO11模型的荔枝图像分类方法,包括:采用荔枝图像数据库对YOLO11模型进行训练,其中所述YOLO11模型包括主干网络、颈部网络和决策网络;其中所述主干网络包括卷积层和局部注意力模块HaloAttention,所述卷积层包括轻量级自适应提取卷积LAE;所述颈部网络包括CARAFE特征采样块;采用训练好的YOLO11模型进行荔枝品种预测。本申请在YOLO11模型的主干网络中引入局部注意力HaloAttention,提升局部范围内特征表达能力;用轻量级自适应提取卷积LAE替换Conv卷积,强化通道间的特征互补;引入CARAFE特征采样块,根据输入特征动态调整采样过程;本发明能够准确的预测荔枝果实的品种,具有快速识别、识别准确率高以及稳定性强的优势。
技术关键词
图像分类方法
注意力
网络
执行存储器存储
荔枝果实
上下文特征
邻域
动态
模块
图像分割
处理器
终端
决策
训练集
通道
编码器
参数
规模