一种基于改进UNet模型的烟叶叶脉分割方法及装置

本发明提出一种基于改进UNet模型的烟叶叶脉分割方法及装置，属于烟叶分类与识别技术领域。烟叶叶脉分割方法包括：采集烟叶样品的高光谱图像并构建烟叶高光谱图像的样本集；构建基于UNet网络的叶脉分割模型，叶脉分割模型至少包括编码器、空间注意模块和解码器；构建生成式对抗网络模型，将样本集输入叶脉分割模型并利用生成式对抗网络模型训练叶脉分割模型；通过训练好的叶脉分割模型对待测烟叶的高光谱图像进行叶脉分割。本发明提出的烟叶叶脉分割方法及装置，在不同条件下均能保持稳定的分割性能，为烟叶品质评价和加工提供了高效的技术解决方案，显著提升了分割精度和鲁棒性。