基于改进UNet网络的金丝球焊直径测量研究

作者:张杰; 唐立新*; 陈子章; 安成
来源:现代制造工程, 2022, (05): 110-114.
DOI:10.16731/j.cnki.1671-3133.2022.05.017

摘要

提出了一种基于改进UNet网络的金丝球焊焊点精确分割与测量的新方法。改进UNet网络由编码器和解码器两部分构成,其中编码器主要用来提取图像特征,使用在ImageNet数据集上预训练分类网络的卷积模块权重初始化编码器部分,可以在不增加训练数据的情况下,加速网络训练且避免过拟合;解码器主要是结合深层特征和浅层特征以实现精确分割,使用改进的多尺度卷积模块替换原网络中的单尺度卷积模块,使解码器能综合利用不同感受野的特征,进一步提升网络的分割精度。实验结果表明,改进UNet网络与原始UNet网络相比,其测试集分割交并比和F分数分别提升了2.04%和1.58%,且直径测量平均误差从7.734μm降低到1.435μm,满足实际检测需求。

全文