摘要
本发明涉及图像处理技术领域,具体涉及语义分割模型的训练方法、语义分割方法及装置。包括获取包含树木的图像并预处理,构建图像集;基于U‑Net架构进行图像分割,U‑Net架构包括编码器、解码器、跳跃连接;基于Dice系数损失和加权交叉熵损失确定总损失函数优化图像背景和树木类别不平衡;通过Adam优化器结合梯度的一阶矩估计和二阶矩估计以在每个训练步骤中调整学习率,并结合余弦退火调度实现Adam优化器学习率衰减调整;基于配置的Adam优化器和余弦退火调度实现语义分割模型的训练,直至收敛得到语义分割模型。由此基于得到的语义分割模型能够便于后续于教育机器人中进行图像的分割处理应用。