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_greenupper_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)

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

相关文章