工业缺陷实例分割方法
摘要:
本发明公开了一种工业缺陷实例分割方法。所述方法包括以下步骤:训练时不固定特征提取网络的浅层网络参数,以保证在开源自然实例数据集上得到的预训练网络能更好地拟合到工业缺陷实例,对工业缺陷实例进行有效的特征提取;使用级联缺陷区域建议模块,避免了从自然实例分割任务迁移到工业缺陷实例分割任务进行的大量参数调整,采用了多次候选框优化的机制提高了算法精度;提出了自适应分割质量评估探头,该模块能根据分割结果生成自适应的分割阈值进行分割结果的二值化,并对分割结果进行打分,解决了实例分割打分偏高的问题。以上设计使得本发明能够有效地解决工业缺陷实例分割问题。
0/0