[英]How to convert a single BGR array to HSV array in Python with OpenCV?
[英]python - opencv - convert pixel from bgr to hsv
img = cv2.imread('example.jpg')
img_hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
# lower mask (0-10)
lower_red = np.array([0, 50, 50])
upper_red = np.array([10, 255, 255]
mask0 = cv2.inRange(img_hsv, lower_red, upper_red)
# upper mask (170-180)
lower_red = np.array([170, 50, 50])
upper_red = np.array([180, 255, 255])
mask1 = cv2.inRange(img_hsv, lower_red, upper_red)
# join my masks
mask = mask0 + mask1
height = mask.shape[0]
width = mask.shape[1]
# iterate over every pixel
for i in range(height):
for j in range(width):
px = mask[i,j]
print px
# check if pixel is white or black
if (px[2] >= 0 and px[2] <= 40):
在上面的例子中,'px' 是 BGR 中的一个像素。 我需要将值转换为 HSV,因为我想检查像素是否在某个颜色范围内。
我已经试过了
colorsys.rgb_to_hsv(px[2], px[1], px[0})
这会引发错误:标量变量的无效索引
谢谢!
从文档:
# Convert BGR to HSV
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
# define range of blue color in HSV
lower_blue = np.array([110,50,50])
upper_blue = np.array([130,255,255])
# Threshold the HSV image to get only blue colors
mask = cv2.inRange(hsv, lower_blue, upper_blue)
您可以使用内置方法将整个img
转换为 hsv:
hsv_img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
这个功能对我有用:
def convert_rgb_to_yuv(frame):
"""
Convert a given rgb image into hsv image
:param frame: Color image to convert
:return: YUV image as numpy array
"""
# CODE HERE
#Conversion matrix from rgb to yuv, transpose matrix is used to convert from yuv to rgb
yuv_from_rgb = np.array([[0.114, 0.587, 0.299],
[0.436, -0.28886, -0.14713],
[-0.10001, -0.51499, 0.615]])
# do conversion
image = frame.dot(yuv_from_rgb.T)
# add the constants based on the conversion formula
image += np.array([16, 128, 128]).reshape(1, 1, 3)
# convert the image to uint8 format
image = np.array(image, dtype = "uint8")
return image
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.