一种基于改进U-Net的葡萄根系图像分割方法
摘要:
本发明公开了一种基于改进U‑Net的葡萄根系图像分割方法,涉及植物图像技术领域,通过获取待分割的葡萄根系图像,并标注;将标注后的葡萄根系图像转化成数据集,将所述数据集分为:训练集、验证集和测试集;构建改进的U‑Net网络,改进的U‑Net网络中设有级联扩张卷积模块;通过训练集、验证集对改进的U‑Net网络进行训练、验证得到最终U‑Net网络;将测试集进行预处理,将预处理后的测试输入所述最终U‑Net网络,通过最终U‑Net网络输出相应的根系分割图像。使用改进的U‑Net网络,加入级联扩张卷积模块,能够扩大网络感受野,增加葡萄根系细节分割效果,降低干扰因素对分割的影响。
0/0