使用OpenCV分割彩色图像

使用OpenCV分割彩色图像

算法原理

  1. 色彩空间转换

使用cv.cvtColor 函数将输入的图像从BGR(蓝绿红)色彩空间转换到HSV色彩空间。HSV色彩空间是一种更符合人眼对颜色感知的模型,其中H(Hue)表示色调,S(Saturation)表示饱和度,V(Value)表示亮度。

  1. 定义颜色范围

在HSV色彩空间中,通过设定色调(H)、饱和度(S)和亮度(V)的上下限来定义一个颜色范围。在这个例子中,设定的是蓝色的范围。lower_hsv和upper_hsv两个数组分别表示蓝色范围的下限和上限

  1. 二值化转换

使用cv.inRange函数将HSV图像二值化。这个函数会遍历图像的每一个像素,检查其HSV值是否在lower_hsv和upper_hsv定义的范围内。如果在范围内,则该像素的值为255(白色),否则为0(黑色)。这样就得到了一个只包含目标颜色(蓝色)的二值化图像,通常被称为掩码(mask)。

  1. 提取颜色部分

使用cv.bitwise_and函数将二值化掩码与原始BGR图像进行"与"操作。这个操作会将原始图像中不在蓝色范围内的部分(即掩码为黑色的部分)设置为黑色,而在蓝色范围内的部分(即掩码为白色的部分)保持不变。这样就得到了一个只包含蓝色部分的图像。

效果

原图:

进行分割后的图:

代码

python 复制代码
import cv2 as cv
import numpy as np


def color_seperate(image):
    hsv = cv.cvtColor(image, cv.COLOR_BGR2HSV)  # 对目标图像进行色彩空间转换
    lower_hsv = np.array([100, 43, 46])  # 设定蓝色下限
    upper_hsv = np.array([124, 255, 255])  # 设定蓝色上限

    # 依据设定的上下限对目标图像进行二值化转换
    mask = cv.inRange(hsv, lowerb=lower_hsv, upperb=upper_hsv)

    # 将二值化图像与原图进行"与"操作;实际是提取蓝色部分
    dst = cv.bitwise_and(image, image, mask=mask)  # 注意:这里将src替换为image

    return dst


# 导入目标图像,获取图像信息
src = cv.imread('blue_1.jpeg')
if src is not None:  # 确保图像被正确读取
    # 调用函数并显示结果
    dst = color_seperate(src)
    cv.imshow('image', src)  # 显示原始图像
    cv.imshow('result', dst)  # 显示处理后的图像
    cv.waitKey(0)  # 等待按键
    cv.destroyAllWindows()  # 销毁所有窗口
else:
    print("Error: Unable to load image.")
相关推荐
刀客12320 分钟前
python3+TensorFlow 2.x(四)反向传播
人工智能·python·tensorflow
SpikeKing26 分钟前
LLM - 大模型 ScallingLaws 的设计 100B 预训练方案(PLM) 教程(5)
人工智能·llm·预训练·scalinglaws·100b·deepnorm·egs
小枫@码1 小时前
免费GPU算力,不花钱部署DeepSeek-R1
人工智能·语言模型
liruiqiang051 小时前
机器学习 - 初学者需要弄懂的一些线性代数的概念
人工智能·线性代数·机器学习·线性回归
Icomi_1 小时前
【外文原版书阅读】《机器学习前置知识》1.线性代数的重要性,初识向量以及向量加法
c语言·c++·人工智能·深度学习·神经网络·机器学习·计算机视觉
微学AI1 小时前
GPU算力平台|在GPU算力平台部署可图大模型Kolors的应用实战教程
人工智能·大模型·llm·gpu算力
西猫雷婶1 小时前
python学opencv|读取图像(四十六)使用cv2.bitwise_or()函数实现图像按位或运算
人工智能·opencv·计算机视觉
IT古董1 小时前
【深度学习】常见模型-生成对抗网络(Generative Adversarial Network, GAN)
人工智能·深度学习·生成对抗网络
Jackilina_Stone1 小时前
【论文阅读笔记】“万字”关于深度学习的图像和视频阴影检测、去除和生成的综述笔记 | 2024.9.3
论文阅读·人工智能·笔记·深度学习·ai
远洋录1 小时前
AI Agent的安全实践:权限控制与数据保护
人工智能·ai·ai agent