源码来了 | 秒杀 DeepLabV3+ 和 Unet 的 缺陷检测网络

点击上方“AI算法与图像处理”，选择加"星标"或“置顶”

重磅干货，第一时间送达

推荐文章【点击下面可直接跳转】：

来源： opencv学堂

下载源码请star这个项目一波

复现测试

上次写了一篇文章介绍CVPR 2019最新提交的工业缺陷检测新思路基于图像语义分割网络实现缺陷检测，当时我们的一位读者看到非常感兴趣，关键是还很厉害，直接实现了论文中提到缺陷检测网络，基于tensorflow+slim框架复现了基于KolektorSDD数据集的检测效果，先看一下测试运行效果：

框架、数据集、代码运行

开发框架与语言版本

• python 3.6

• cuda 9.0

• cudnn 7.1.4

• Tensorflow 1.12

 

数据集下载地址

https://www.vicos.si/Downloads/KolektorSDD

 

复现论文的时候作者并没有完全按照论文来设置参数，但是仍然取得了比较好的训练与测试效果，KolektorSDD前面30子文件夹作为训练数据，后面20个作为测试数据，最终的准确率与召回如下：

 

直接运行如下命令即可加载ckpt文件完成测试

python run.py –test

 

单独训练分割网络

python run.py --trian_segment

 

单独训练决策网络

python run.py  --train_decison

 

分割与决策网络一起训练

python run.py  --train_total

 

最后看一下测试运行脚本与结果

测试结果如下：

源码下载

源码下载地址，别忘记star这个项目！

https://github.com/Wslsdx/Deep-Learning-Approach-for-Surface-Defect-Detection

有热门推荐

加群交流

欢迎小伙伴加群交流，目前已有交流群的方向包括：AI学习交流群，目标检测，秋招互助，资料下载等等；加群可扫描并回复感兴趣方向即可

你点的每个“在看”，我都认真当成了喜欢