Python Opencv提取图片中某种颜色组成的图形的方法

下面是针对“Python Opencv提取图片中某种颜色组成的图形的方法”的完整攻略：

下面是针对“Python Opencv提取图片中某种颜色组成的图形的方法”的完整攻略：

准备工作

首先需要安装OpenCV库，可以使用以下命令进行安装：

pip install opencv-python

在Python代码中，需要用到以下几个包：

import cv2
import numpy as np

方法一：利用颜色空间转换

1. 将图像转换为HSV颜色空间

img = cv2.imread('test.jpg')
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

1. 定义要提取的颜色范围，通过调节下列代码的lower_green与upper_green的值，可以提取不同颜色的图形

lower_green = np.array([35, 43, 46])
upper_green = np.array([77, 255, 255])
mask = cv2.inRange(hsv, lower_green, upper_green)

1. 对二值化图像进行处理

res = cv2.bitwise_and(img, img, mask=mask)

方法二：利用形态学处理

该方法与方法一有点不同，将不会进行颜色空间转换

1. 对原图像分离B、G、R三个通道

img = cv2.imread('test.jpg')
b, g, r = cv2.split(img)

1. 对绿色通道进行处理

_, green_th = cv2.threshold(g, 90, 255, cv2.THRESH_BINARY)
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (3,3))
morphology_result = cv2.erode(green_th, kernel=kernel, iterations=1)

1. 对二值化图像进行处理

res = cv2.bitwise_and(img, img, mask=morphology_result)

以上是两种提取图片中某种颜色组成的图形的方法，可以根据需求选择合适的方法进行实现。