基于U-net的香菇表型提取方法

摘要

针对香菇表型测量信息化程度低、人工测量费时费力的问题，提出了一种基于U-Net的香菇表型提取方法。通过采集香菇切面图像，建立数据集，实现了基于U-Net的香菇菌盖、菌柄、左右菌褶的分割，模型的平均交并比和平均像素准确率分别为85.00%和91.25%。结合质心法和最小外接矩形法实现了菌盖直径、菌盖厚度、菌柄长度、菌柄直径和菌褶宽度5个香菇表型参数的自动测量。与人工测量值相比，提出的方法在测量菌盖直径、菌盖厚度、菌柄长度、菌柄直径和菌褶宽度时，其平均绝对百分比误差分别为1.57%、5.01%、2.57%、5.47%、2.74%；均方根误差分别为0.12、0.08、0.09、0.10、0.06 cm；决定系数均大于0.90。结果表明，基于U-Net的香菇表型提取方法适用于香菇的表型测量，具备较高的分割准确率。

关键词

香菇 / 图像分割 / 表型提取 / U-net