基于改进的U-net语义分割网络的补强胶定位方法

摘要

本发明公开了一种基于改进的U‑net语义分割网络的补强胶定位方法，采集大量补强胶图片，制作成训练样本；输入训练图像首先经过第一编码器得第一特征图；第一特征图经最大池化层后大小减小一半，经过第二编码器、注意力模块和金字塔池化模块得第二特征图；第二特征图经最大池化层后大小减小一半，经第三编码器得到第三特征图；将第三特征图使用解码器模块与第二特征图相融合后将特征图大小上采样两倍得第一融合特征图，第一融合特征图经卷积层后使用解码器模块与第一特征图相融合后再将特征图大小上采样两倍得第二融合特征图，第二融合特征图经卷积层得到分割结果。该方法具有提高机器视觉算法应用在补强胶定位中普适性的优点。