【python】OpenCV—Color Detection

学习来自 如何使用 OpenCV Python 检测颜色

py 复制代码
import cv2
import numpy as np


def red_hsv(img, save=False):
    lower_hsv1 = np.array([0, 175, 20])
    higher_hsv1 = np.array([10, 255, 255])

    lower_hsv2 = np.array([170, 175, 20])
    higer_hsv2 = np.array([10, 255, 255])

    mask1 = cv2.inRange(img, lower_hsv1, higher_hsv1)
    mask2 = cv2.inRange(img, lower_hsv2, higer_hsv2)

    if save:
        cv2.imwrite("red-mask-1.jpg", mask1)
        cv2.imwrite("red-mask-2.jpg", mask2)
        cv2.imwrite("red-mask.jpg", mask1+mask2)

    return mask1 + mask2


def green_hsv(img, save=False):
    lower_hsv1 = np.array([40, 150, 20])
    higher_hsv1 = np.array([70, 255, 255])

    mask1 = cv2.inRange(img, lower_hsv1, higher_hsv1)
    if save:
        cv2.imwrite("green-mask-1.jpg", mask1)

    return mask1


if __name__ == "__main__":
    bgr = cv2.imread("1.png")
    hsv = cv2.cvtColor(bgr, cv2.COLOR_BGR2HSV)
    cv2.imwrite("hsv.jpg", hsv)

    mask = red_hsv(hsv, save=True) + green_hsv(hsv, save=True)
    cv2.imwrite("mask.jpg", mask)

    result = cv2.bitwise_and(bgr, bgr, mask=mask)
    cv2.imshow("color detection", result)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

转成 hsv 色域,来过滤掉相应的颜色

H 根据 x 轴取值,S 根据 y 轴取值,V 始终在 (20-255) 之间的范围内取值

bitwise_and,输出图像像素只有mask对应位置元素不为0的部分才输出,否则该位置像素的所有通道分量都设置为0

看看结果

输入图片

hsv

red-mask-1

red-mask-2

red-mask

green-mask-1

result

相关推荐
honeysuckle_luo44 分钟前
RandLA-net-pytorch 复现
人工智能·pytorch·python
SunnyDays10112 小时前
Python 高效实现 Excel 与 TXT 文本文件之间的数据转换
python·excel转txt·文本转excel·excel转文本·txt转excel
一念&2 小时前
每日一个C语言知识:C 头文件
c语言·开发语言·算法
硬件人某某某3 小时前
python基于卷积神经网络的桥梁裂缝检测系统(django),附可视化界面,源码
python·cnn·django
DARLING Zero two♡3 小时前
仓颉GC调优参数:垃圾回收的精密控制艺术
开发语言·仓颉
今日说"法"3 小时前
Rust探秘:所有权转移在函数调用中的表现
开发语言·后端·rust
java1234_小锋3 小时前
PyTorch2 Python深度学习 - 自动微分(Autograd)与梯度优化
开发语言·python·深度学习·pytorch2
Python私教3 小时前
C 语言运算符全景:从入门到进阶
c语言·开发语言·网络
java1234_小锋3 小时前
PyTorch2 Python深度学习 - 简介以及入门
python·深度学习·pytorch2
csbysj20205 小时前
Perl 格式化输出
开发语言