基于AI为产品质检赋能已经产生了许多有意义的应用,在我前面的一些文章里面也有助力产品生产质检相关的项目,但大都是基于检测模型开发的,这里主要是想基于图像分割的方式来实现产品质检,本文主要是基于DeepLabV3Plus模型实现划分分割,首先看下效果图:

原始样例数据如下:

划痕相对来说还是比较明显的。
为了直观对比,这里对标注数据进行了可视化展示如下所示:

这里的划痕线条相对比较粗,所以模型还是比较好识别的。
我使用的是开源的项目,在这里,首页截图如下所示:

已有1.1k的star量,不过这个看自己需要了,可以根据实际需要选择合适的项目即可。



上面是官方给出来的实例效果图。
下面我基于自己训练出来的模型进行测试如下:

结果如下:



直观上来看分割的效果还是比较细腻的。感兴趣的话可以自己试试看。