OpenCV静态图像分割(抠图)

很多时候,我们需要对静态图像进行分割,将前景与背景分离。OpenCV原生提供了一些图像分割的方法。

我们主要以下图为例,测试图像分割的效果。我们期待将图中的苹果抽离出来。图中含有高光、阴影,这是分割的难点。

一、漫水填充

此算法以一个种子点为中心,将颜色相近的像素修改为目标颜色。Photoshop中使用的魔术棒工具就是使用此算法。主要参数是种子点的位置和颜色变化上下限。此算法只适用于背景特别单调,前景边缘清晰的场合。即便如此,受环境影响,参数也很难调。一般需要人工辅助。

测试结果:

二、分水岭算法

此算法需要用户标记出不同区域,算法将周边相似的像素合并到区域中。对于测试图,我们把前景(苹果)标记为一个区域,将背景标记为另一个区域,标记图如下:

算法运行的结果如下:

三、Grabcuts算法

此算法需要用户画出一个矩形,矩形内为前景,外为背景,算法求出具体前景范围。算法因为需要迭代,运行速度较慢。

下图显示出标记的矩形区域,以及计算出的前景(苹果)具体范围。

四、Mean-Shift算法

此算法并不是真正的图像分割,只是去除图像中的精细纹理,减少颜色梯度。可以作为一种预处理算法,后续使用其他算法进行分割。

算法运行的结果如下,接近于一种模糊:

五、K-Means算法

此算法将图像分为K个类簇,需预先确定有多少个类簇。类簇的数量对分割的效果有重大影响。

设置2个类簇的结果:

设置3个类簇的结果:

六、U2-Net

以上,便是OpenCV原生的图像分割方法。可以看到,如果没有人工干预,效果交不理想,即使有人工干预,效果也不太理想(阴影没有办法处理)。

U2-Net是一个优秀的图像分割模型,能够提出中图片中的主要目标。

模型有多个版本,u2net、u2netp、u2net_human_seg等,可根据实际情况挑选模型。

模型的运行速度和精度都很不错。

以下是算法处理结果:

可以看到,苹果的范围被很好的标记出来,甚至阴影也处理好了。

相关推荐
动物园猫10 小时前
PCB表面缺陷目标检测数据集:6类别、3,500张图像 | 目标检测
人工智能·目标检测·计算机视觉
Byron Loong11 小时前
【视觉】UVC 协议用于工业的优势
计算机视觉
深度研习笔记11 小时前
OpenCV工业视觉进阶14|SQLite违规数据库存储+历史记录查询+Excel报表导出,完成项目商用数据闭环(验收必备)
数据库·opencv·sqlite
深度研习笔记1 天前
OpenCV工业视觉实战09|项目EXE打包+工控机无环境部署+后台常驻运行,彻底脱离Python环境,完成项目最终交付
python·opencv·webpack
深度研习笔记1 天前
OpenCV工业视觉实战06|ROI区域截取+局部精细化检测,锁定作业区域、屏蔽无效背景,大幅提升YOLO检测准确率
人工智能·opencv·yolo
fai厅的秃头姐!1 天前
OpenCV-基础篇
人工智能·opencv·计算机视觉
丨白色风车丨1 天前
OpenCV 基于 HSV 颜色空间实时颜色分割(摄像头蓝色识别)
人工智能·opencv·计算机视觉·hsv颜色识别·python计算机视觉
深度研习笔记1 天前
OpenCV工业视觉实战10|违规语音播报+弹窗告警+重复告警过滤,打造智能声光预警系统(项目加分必备)
人工智能·opencv·计算机视觉
AI人工智能+1 天前
银行回单识别技术通过AI驱动的智能文档理解方案,实现财务处理的革命性升级
深度学习·计算机视觉·自然语言处理·ocr·银行回单识别
hhzz1 天前
CNN图像分类入门:基于TensorFlow+Keras的CIFAR-10数据集全流程实战
图像处理·计算机视觉·ai·cnn·大模型