python opencv 检测特定颜色

原创 2016年08月28日 16:54:01

15-python opencv 检测特定颜色


概述

本节实现的是使用OpenCV里自带的函数,检测出视频里图像中的蓝色和天蓝色、青色,比如我的手机背景、我衣服上的蓝色格子,墙砖的蓝色和学生证照片上的蓝色背景。

  • 设置“蓝色”HSV范围
  • BGR转HSV
  • 捕获摄像头视频
  • 获取蓝色部分mask
  • 图像按位与操作
  • 显示蓝色部分图像

实现过程

定义视频对象

视频对象用于捕获摄像头视频流。

import cv2
import numpy as np

cap = cv2.VideoCapture(0)

设置HSV中蓝色、天蓝色范围

这里主要参考了这个博客,我设置的颜色范围如下。

# set blue thresh
lower_blue=np.array([78,43,46])
upper_blue=np.array([110,255,255])

获取视频帧并转成HSV格式

利用cvtColor()将BGR格式转成HSV格式,参数为cv2.COLOR_BGR2HSV。

# get a frame and show
ret, frame = cap.read()
cv2.imshow('Capture', frame)
# change to hsv model
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

获取mask

利用inRange()函数和HSV模型中蓝色范围的上下界获取mask,mask中原视频中的蓝色部分会被弄成白色,其他部分黑色。

# get mask
mask = cv2.inRange(hsv, lower_blue, upper_blue)
cv2.imshow('Mask', mask)

图像按位操作

将mask于原视频帧进行按位与操作,则会把mask中的白色用真实的图像替换:

# detect blue
    res = cv2.bitwise_and(frame, frame, mask=mask)
    cv2.imshow('Result', res)

源代码

程序的源代码如下:

# created by Huang Lu
# 28/08/2016 14:46:31 
# Department of EE, Tsinghua Univ.

import cv2
import numpy as np

cap = cv2.VideoCapture(0)

# set blue thresh
lower_blue=np.array([78,43,46])
upper_blue=np.array([110,255,255])

while(1):
    # get a frame and show
    ret, frame = cap.read()
    cv2.imshow('Capture', frame)

    # change to hsv model
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    # get mask
    mask = cv2.inRange(hsv, lower_blue, upper_blue)
    cv2.imshow('Mask', mask)

    # detect blue
    res = cv2.bitwise_and(frame, frame, mask=mask)
    cv2.imshow('Result', res)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

也可以参考我的GitHub上的,点击这里

运行结果

在命令行进入该源程序所在目录后,运行python main.py后即可显示结果。显示结果如下:

结果

可以发现,我的手机背景、我衣服上的蓝色格子,墙砖的蓝色和学生证照片上的蓝色背景都被正确的识别出来了,但是还存在一些噪声,有待进一步改进。

参考

版权声明:本文为博主原创文章,未经博主允许不得转载。

《python+opencv实践》一、基于颜色的物体追踪(上)

本文主要参考国外一大牛博客,然后自己修改得来。 相关知识点在这里 实现功能:追踪红颜色瓶盖,并画出瓶盖轮廓和运动轨迹。 from collections import deque import nu...

Delphi7高级应用开发随书源码

  • 2003年04月30日 00:00
  • 676KB
  • 下载

利用python打开摄像头及颜色检测

python-opencv实现颜色检测

python opencv检测目标颜色

# -*- coding:utf-8 -*- __author__ = 'kingking' __version__ = '1.0' __date__ = '14/07/2017' import cv...

OpenCV颜色识别

彩色模型数字图像处理中常用的采用模型是RGB(红,绿,蓝)模型和HSV(色调,饱和度,亮度),RGB广泛应用于彩色监视器和彩色视频摄像机,我们平时的图片一般都是RGB模型。而HSV模型更符合人描述和解...
  • zwhlxl
  • zwhlxl
  • 2015年06月05日 20:24
  • 6905

Delphi7高级应用开发随书源码

  • 2003年04月30日 00:00
  • 676KB
  • 下载

Python 3 色情图片识别

Python 3 色情图片识别 一、实验简介 本实验将使用 Python3 去识别图片是否为色情图片,我们会使用到 PIL 这个图像处理库,会编写算法来划分图像的皮肤区域 1.1. ...

opencv颜色过滤--------如何选择需要的颜色呢?

在日常生活中我们表示颜色的时候都喜欢用RGB模型进行表示,RGB分别代表了三原色:红色Red, 绿色Green,蓝色Blue。但是当我们想要从图片中选取某种颜色的时候,比如说红色,用RGB该怎么做?很...

Python中颜色处理

本节所讲内容:作业脚本 作业: 实战:创建一个商品页面,要求 1、显示商品名称和价格对应关系 2、获取用户工资水平 3、提供用户选择购买商品功能 4、用户购买完商品,显示用户账号余额 5...

Delphi7高级应用开发随书源码

  • 2003年04月30日 00:00
  • 676KB
  • 下载
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:python opencv 检测特定颜色
举报原因:
原因补充:

(最多只允许输入30个字)