改进YOLOv4-tiny算法的焊缝特征点提取方法

    公开(公告)号:CN115546604A

    公开(公告)日:2022-12-30

    申请号:CN202211115696.3

    申请日:2022-09-14

    Abstract: 本发明公开了改进YOLOv4‑t iny算法的焊缝特征点提取方法,其特征在于:包括S1建立焊缝图像数据集,其中将90%的焊缝图像数据集作为训练集,将余下10%的焊缝图像数据集作为测试集;S2利用训练集对改进后的YOLOv4‑Tiny网络模型进行训练;所述改进后的YOLOv4‑Tiny网络模型包括一条与YOLOv4‑t iny算法浅层卷积层相连的路径,以提高焊缝特征提取能力;S3在训练全部结束后,将训练好的权值参数保存到相应文件中,以便用于对测试集的图片进行焊缝目标的检测;S4利用测试集对经过训练的改进后的YOLOv4‑Tiny网络模型进行焊缝目标的检测。在YOLOv4‑t iny算法的浅层加入有效的特征提取层,提取更多的焊缝特征点特征信息,从而减少了检测误差。

Patent Agency Ranking