一种基于模型压缩的钢铁产品表面缺陷检测方法

摘要

本发明公开了一种基于模型压缩的钢铁产品表面缺陷检测方法，属于钢材表面缺陷检测技术领域，包括以下步骤：S1：对热轧带钢表面缺陷图像样本进行预处理；S2：基于anchor‑free目标检测框架，搭建热轧带钢表面缺陷检测网络；S3：对缺陷检测网络进行模型压缩，并采用线性策略及训练样本训练模型；S4：在独立验证集上测试缺陷检测网络并输出检测速度与检测结果。本发明先搭建用于识别与定位的钢铁表面缺陷检测网络，再移除缺陷检测网络中冗余的参数，减少模型高额的存储空间以及计算资源消耗，将缺陷检测网络更稳定、高效的落地在硬件平台上。