摘要
本发明公开了一种基于卷积神经网络的烟叶颜色校正方法,包括:1、采集仓库烟叶图像并进行归一化、去噪预处理,同步获取标准光照下的颜色参考值作为监督标签;2、构建融合SqueezeNet轻量级网络与SE通道注意力机制的特征提取模型,增强颜色特征适应能力;3、以预处理图像为输入,采用监督学习策略通过损失函数优化迭代训练模型,提升颜色映射精度;4、应用训练模型对烟叶图像进行光照校正,输出标准光照下的还原图像。本发明通过轻量化网络降低计算成本,利用注意力机制强化关键特征学习,并结合实际场景数据与标准光照标定的双重优化,能显著提高颜色校正准确性,可有效应用于烟叶质量检测及仓储管理中的自动化颜色分析。
技术关键词
烟叶图像
烟叶颜色
阶段
校正方法
特征提取模块
颜色校正
监督学习策略
多层网络结构
通道
注意力机制
损失函数优化
更新网络参数
特征提取模型
全局平均池化
级联
光照特征