基于改进SOLOv2的穴盘幼苗图像分割方法

作者：庄前伟; 王志明^*; 吴龙贻; 李恺; 王春辉

来源：南京农业大学学报, 2023, 46(01): 200-209.

• 图像分割
• 穴盘幼苗
• 改进SOLOv2
• 分割精度
• 实时

[目的]针对传统图像分割算法对穴盘幼苗图像分割精确度不高、分割效率低等问题，本研究提出一种基于改进SOLOv2的穴盘幼苗实例分割算法。[方法]在SOLOv2算法架构的基础上，将拆分注意力网络(ResNeSt)与特征金字塔(feature pyramid network, FPN)相融合作为主干提取网络，提取幼苗的特征信息；采集穴盘幼苗图像制作成数据集，引入多尺度变换和网格擦除等手段进行数据增强，再结合可变形卷积(DCNv2)优化网络结构，提升分割的精确度；引入秩和排序损失(Rank&Sort Loss)解决类别分支和掩码分支任务中超参数调节存在非最优和耗时的问题，提高SOLOv2分割的性能和效率。[结果]基于采集多变机械手抓取穴盘幼苗数据集的试验结果表明，改进SOLOv2分割算法的平均精度均值(mean average precision, mAP)、交并比0.5时平均精度(AP50)和交并比0.75时平均精度(AP75)分别为0.641、0.885和0.657,每张幼苗图像在GPU(graphic processing unit)上的平均分割时间为0.114 s,比原SOLOv2的各项精度指标分别高2.8%、3.1%和2.3%,分割时间减少0.058 s。[结论]改进SOLOv2算法具有更强的泛化性和鲁棒性，可以实现果蔬幼苗的实时高精度分割，并为幼苗品质分级筛选提供技术帮助。

• 单位
  南京理工大学