卷积神经网络可视化在流程工业图像识别中的应用

摘要

流程工业图像识别中常常无法直观地知道被训练图像所关注的重点是否是想要模型学习的内容,因而无法确定训练模型是否符合初始训练的需求.基于Keras的卷积神经网络可视化方法来了解图像分类问题中图像哪些部分起到了关键作用,同时定位图像中物体的位置,可为后续模型训练提供参考依据.对工业现场的图像数据进行类激活热力图可视化处理后,再利用选定的网络模型进行训练,通过该方法开发的识别模型在最终测试集上达到了98.6％的正确率.