一种基于YOLOX的钢材表面缺陷检测方法、系统、设备及介质

本发明公开了一种基于YOLOX的钢材表面缺陷检测方法、系统、设备及介质，涉及钢材检测技术领域，包括采集钢材表面缺陷数据集，并进行预处理和标注后，将数据集划分为训练集和测试集；对YOLOX网络结构模型进行改进，搭建改进的YOLOX网络结构模型；使用所述训练集对改进的YOLOX网络模型进行训练，得到钢材表面缺陷检测模型；使用所述测试集对钢材表面缺陷检测模型进行测试。本发明使用YOLO系列算法，改进主流检测算法中会预测出很多冗余的预测框的通病，在YOLOX算法的基础上引入了ASFF模块，提高模型的特征融合能力，更好地进行缺陷检测，使用改进的SE注意力机制，提高模型的特征提取能力，采用EIOU损失函数改善模型定位不准确的问题，加强了识别能力获得更高的检测精度。
上海电力大学
蒋三新 | 刘毅 | 韩明桥 | 陈雨 | 朱寅