一种基于改进YOLOX-S算法的金属铸件表面缺陷检测方法

    公开(公告)号:CN115797326A

    公开(公告)日:2023-03-14

    申请号:CN202211678195.6

    申请日:2022-12-26

    Applicant: 安徽大学

    Abstract: 本申请提供一种基于改进YOLOX‑S算法的金属铸件表面缺陷检测方法,步骤为:1)采集数据库中若干金属铸件表面图像,构建金属铸件表面数据集;2)对金属铸件表面数据集中的数据进行预处理,并将处理后数据集按比例划分训练集和测试集;3)搭建基于改进YOLOX‑S网络的金属铸件表面缺陷检测网络;4)利用数据集训练基于改进YOLOX‑S算法的金属铸件表面缺陷检测网络;5)将待检测的金属铸件表面图像输入训练好的改进YOLOX‑S网络中,获得含有缺陷类别和缺陷位置的结果图。本申请针对于金属铸件表面缺陷构建基于改进YOLOX‑S算法的金属铸件表面缺陷检测模型,通过擦除机制和加强机制有效地解决了了原始YOLOX‑S网络存在的语义差异问题,提高了网络对于密集缺陷的检测精度。

Patent Agency Ranking