基于改进注意力机制的薄膜瑕疵检测方法
摘要:
本发明公开了一种基于改进注意力机制的薄膜瑕疵检测方法。本发明首先实时读取薄膜图像;然后将图像输入到网络模型中进行推理;再通过推理结果判断薄膜图像中是否有瑕疵,如有则自动用红的矩形框标注,否则读取下一张薄膜图像。本发明通过对YOLOv5网络结构的简化修改,通过对网络加入轻量级的注意力模块,使得在不影响速度的前提下能有效提高瑕疵检测的准确率;该方法用于工业薄膜瑕疵检测可以有效提高薄膜产品的质量,且不需要人工干预,节省了人工及时间成本。
公开/授权文献
0/0