U-Net实现缺陷检测( 二 )


最后调用model.fit()来进行U-Net模型的训练,由于是在CPU下运行,因此时间较长,整个训练过程持续大约18个小时 。
训练结果
获得loss和输出图像如下:
在训练完成之后保存模型为.h5,从测试集中随机取出部分图像查看分割效果如下(注意将预测结果图的像素点从0~1还原为0, 255二值分布) 。从左到右依次为:原图,给定的掩膜图和预测分割图 。
参考文献
[1] Evan ,Long,. Fullyfor[M]. IEEE, 2017.
[2]O ,P , Brox T . U-Net:forImage [J]. 2015.
[3]K ,A . Very Deepfor Large-Scale Image [J]., 2014.
[4] He K , Zhang X , Ren S , et al. Deepfor Image [C]// IEEEon&. IEEE, 2016.
[5] 韩慧. 基于深度学习的工业缺陷检测方法研究[D]. 2019.
[6] 刘聪. 基于卷积神经网络的微小零件表面缺陷检测技术研究[D]. 2019.
我一定会来填坑的,确信 。。。