一种用于缺陷检测的概率图可视化方法及装置

摘要

本申请涉及缺陷检测技术领域，涉及一种用于缺陷检测的概率图可视化方法及装置。为了解决现有技术中不能直观显示像素点缺陷类别的概率信息、遗漏背景类别相关信息的问题。所述方法包括：将待检测目标图像输入至语义分割模型，得到与待检测目标图像对应的多通道概率图；根据多通道概率图，获取每个类别中同一像素点的概率，比较得出每个像素点的最大概率；基于每个最大概率对应的类别，记录每个最大概率对应类别的索引号；根据预设类别对应的索引号，判断每个最大概率对应的类别种类；根据类别种类，对每个最大概率进行处理，得到包含每个像素点属于某一类别概率的单通道概率图；将单通道概率图转化为RGB图像，得到伪彩色图，以实现概率图可视化。