基于无监督域适应的跨场景带钢表面缺陷识别 |
| |
引用本文: | 刘坤,杨晓松.基于无监督域适应的跨场景带钢表面缺陷识别[J].浙江大学学报(自然科学版 ),2023(3):477-485. |
| |
作者姓名: | 刘坤 杨晓松 |
| |
作者单位: | 河北工业大学人工智能与数据科学学院 |
| |
基金项目: | 国家自然科学基金资助项目(62173124);;河北省自然科学基金资助项目(F2019202305); |
| |
摘 要: | 深度学习模型面对跨场景的带钢表面缺陷识别时存在泛化性能差的问题,为此提出端到端的多级对齐域适应神经网络模型(MADA),实现源域与目标域数据的像素级光照分布对齐与特征级纹理分布对齐. MADA通过无参考像素级光照分布对齐模块和光照校正损失函数,将源域与目标域数据投影到光照子空间,实现源域与目标域的像素级光照分布对齐.利用纹理特征提取器和特征级域鉴别器的对抗学习,实现源域和目标域数据的纹理分布对齐.实验在邯郸钢铁集团带钢表面缺陷数据集的F1指数达到98%,在谢维尔钢铁集团带钢表面缺陷数据集上的F1指数达到86.6%.实验结果表明,与其他域适应方法相比,所提方法具有更好的泛化性能.
|
关 键 词: | 带钢表面缺陷识别 域适应 跨场景 泛化 光照 纹理 |
|