摘要
本发明属于图像识别技术领域,公开了一种基于三塔式知识蒸馏架构的多类别工业图像异常检测方法及装置,设计了涵盖局部上下文学生网络、全局上下文学生网络和预训练的教师网络的三塔式网络架构。对于用于训练的多类别正常工业图像,通过特征蒸馏损失分别约束两个学生网络输出的特征与教师网络一致,以及通过特征对齐损失约束使用局部和全局上下文区域预测中心特征的两个学生网络之间的输出特征一致,提升模型学习多类别正常图像特征表示的能力,保障模型在联合异常检测场景下对多类别异常样本进行高精度的检测和定位;本发明具备较高的实时性和较小的模型规模,可应用于对实时性要求高的工业异常检测场景。
技术关键词
图像异常检测方法
学生
注意力编码器
蒸馏
网络架构
上下文特征
教师
工业
识别异常图像
多层感知机
计算机可执行程序
输出特征
图像识别技术
像素
可读存储介质
训练集
系统为您推荐了相关专利信息
教育资源推荐
多模态数据采集
知识图谱分析
标签
整合视频观看
前视声呐
蒸馏
学生
焦点损失函数
多尺度特征金字塔
工作站
操作者
人机协同
混合整数规划模型
机器人