[发明专利]一种基于改进U-Net的葡萄根系图像分割方法

权利要求书

1.一种基于改进U-Net的葡萄根系图像分割方法，其特征在于，包括：

获取待分割的葡萄根系图像，并对所述葡萄根系图像进行标注；

将标注后的葡萄根系图像转化成数据集，将所述数据集分为：训练集、验证集和测试集；

构建改进的U-Net网络，所述改进的U-Net网络中设有级联扩张卷积模块；

通过训练集、验证集对所述改进的U-Net网络进行训练、验证得到最终U-Net网络；

将所述测试集进行预处理，将预处理后的测试输入所述最终U-Net网络，通过所述最终U-Net网络输出相应的根系分割图像。

2.根据权利要求1所述的基于改进U-Net的葡萄根系图像分割方法，其特征在于，对所述改进的U-Net网络进行训练、验证之前，对所述改进的U-Net网络进行优化。

3.根据权利要求2所述的基于改进U-Net的葡萄根系图像分割方法，其特征在于，通过对所述改进的U-Net网络的损失函数优化实现对所述改进的U-Net网络进行优化。

4.根据权利要求3所述的基于改进U-Net的葡萄根系图像分割方法，其特征在于，优化所述损失函数采用与加权的损失函数进行优化。

5.根据权利要求1所述的基于改进U-Net的葡萄根系图像分割方法，其特征在于，所述标注为语义分割标注。

6.根据权利要求1所述的基于改进U-Net的葡萄根系图像分割方法，其特征在于，将标注后的葡萄根系图像转化成数据集为：将标注后的葡萄根系图像转换成PASCAL VOC语义分割格式的数据。

7.根据权利要求1所述的基于改进U-Net的葡萄根系图像分割方法，其特征在于，所述改进的U-Net网络包括依次连接的输入层、编码器模块、级联扩张卷积模块、解码器模块和输出层；

所述编码器对输入层输入的葡萄根系图像进行下采样，所述级联扩张卷积模块用于对所述编码器输出的特征图进行卷积扩张，所述解码器模块对经过所述级联扩张卷积模块输出的特征图进行上采样，后通过所述输出层对所述解码器模块输出进行卷积处理得到根系分割图像。