-
公开(公告)号:CN115797326A
公开(公告)日:2023-03-14
申请号:CN202211678195.6
申请日:2022-12-26
Applicant: 安徽大学
IPC: G06T7/00 , G06N3/0464 , G06N3/08 , G06V10/40 , G06V10/82
Abstract: 本申请提供一种基于改进YOLOX‑S算法的金属铸件表面缺陷检测方法,步骤为:1)采集数据库中若干金属铸件表面图像,构建金属铸件表面数据集;2)对金属铸件表面数据集中的数据进行预处理,并将处理后数据集按比例划分训练集和测试集;3)搭建基于改进YOLOX‑S网络的金属铸件表面缺陷检测网络;4)利用数据集训练基于改进YOLOX‑S算法的金属铸件表面缺陷检测网络;5)将待检测的金属铸件表面图像输入训练好的改进YOLOX‑S网络中,获得含有缺陷类别和缺陷位置的结果图。本申请针对于金属铸件表面缺陷构建基于改进YOLOX‑S算法的金属铸件表面缺陷检测模型,通过擦除机制和加强机制有效地解决了了原始YOLOX‑S网络存在的语义差异问题,提高了网络对于密集缺陷的检测精度。