基于U-net模型在烟叶主脉分割中的应用  

Application of U-net Model in Main Vein Segmentation of Tobacco Leaves

作　　者：徐凯成 张伟 XU Kaicheng;ZHANG Wei(Shanghai Weston Information Technology Co.,Ltd.,Shanghai 200052,China)

机构地区：[1]上海威士顿信息技术股份有限公司,上海200052

出　　处：《数字通信世界》2024年第5期129-131,共3页Digital Communication World

摘　　要：烟叶主脉的形状特征是烟叶分级的参考标准之一,该文在U-net模型的基础上进行了改进,并将该模型应用于烟叶表面主脉的提取。该模型在所有上采样层中设置了侧输出,并对这些侧输出进行了约束来提升模型的识别效果。在保证模型性能的同时对U形分割网络进行了剪枝,减少了模型冗余,提升了模型推理速度。对实验结果进行对比分析,验证了该模型在烟叶表面主脉分割任务中的有效性。The shape characteristics of the main veins of tobacco leaves are one of the reference standards for tobacco grading.The article improves on the U-net model and applies it to the extraction of main veins on the surface of tobacco leaves.The model sets side outputs in all upsampling layers and constrains these side outputs to improve the recognition performance of the model.The U-shaped segmentation network was pruned while ensuring model performance,reducing model redundancy and improving model inference speed.The comparative analysis of the experimental results verified the effectiveness of the model in the task of segmenting the main veins on the surface of tobacco leaves.

关 键 词：烟叶主脉 语义分割 U-net 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]