表面瑕疵检测——分割模型和分类模型融合模型决策实现少样本高精度(Segmentation-based deep-learning)

本文介绍了基于分割的表面缺陷检测深度学习方法,通过结合分割网络和决策网络,实现了仅使用25-30个样本的高精度检测。在Keras中构建了网络模型,首先训练分割网络,然后冻结其参数,接着训练决策网络。通过concatenate函数融合两网络,并用KolektorSDD数据库进行数据处理。实际应用证明,该方法在有限样本下达到了满意的效果。
摘要由CSDN通过智能技术生成

这是一篇根据自己的理解,复写论文《Segmentation-Based Deep-Learning Approach for Surface-Defect Detection》代码,也不知道写的对不对,但至少运行结果是自己满意的,达到了较高的准确率。

首先,《基于分割的表面缺陷检测深度学习方法》这篇论文主要是利用分割模型和分类模型进行表面瑕疵的检测,主要优势为:只需要25-30个有缺陷的样本就可完成分类,所用样本极少,具体的论文翻译,直接用google翻译,不太精准但也大致意思了解过,如果需要可留言,这篇博文主要介绍搭建网络的思路。

论文的网络模型:

可以明显得看到主要分为两个网络,Segmentation(分割网络)和Decision(决策网络),这两个网络相互融合。而且论文中也写到了训练的时候先是训练分割网络,而后将其参数freeze(冻结),训练决策网络,完成最终的瑕疵检测。

来看看我的网络思路,用的是keras。

首先是分割网络:

 

搭建就是这样子啦~~

  • 18
    点赞
  • 75
    收藏
    觉得还不错? 一键收藏
  • 38
    评论
评论 38
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值