hsv = cv2.cvtColor(rgb_image, cv2.COLOR_BGR2HSV) 然后利用cv2.inRange函数设阈值,去除背景部分 mask = cv2.inRange(hsv, lower_red, upper_red) 函数很简单,参数有三个 第一个参数:hsv指的是原图
第二个参数:lower_red指的是图像中低于这个lower_red的值,图像值变为0
第三个参数:upper_red指的是图像中高于这个upper_red的值,图像值变为0 而在lower_red~upper_red之间的值变成255
import cv2
img = cv2.imread(r"C:\Users\Curry\Desktop\flower.png")
hsv = cv2.cvtColor(img, code=cv2.COLOR_BGR2HSV)
lower_red = (156, 60, 60)
upper_red = (180, 255, 255)
mask = cv2.inRange(hsv,lower_red,upper_red)
res = cv2.bitwise_and(img,img,mask=mask)
# cv2.imshow('flower', mask)
cv2.imshow('flower', res)
cv2.waitKey(0)
cv2.destroyAllWindows()
原图:
mask:
2.手工绘制轮廓
cv2.fillPoly()函数可以用来填充任意形状的图型.可以用来绘制多边形,工作中也经常使用非常多个边来近似的画一条曲线;cv2.fillPoly()函数可以一次填充多个图型
import cv2
img = cv2.imread(r"C:\Users\Curry\Desktop\flower.png")
hsv = cv2.cvtColor(img, code=cv2.COLOR_BGR2HSV)
h,w,c = img.shape
mask = np.zeros((h, w), dtype=np.uint8)
x_data = np.array([124, 169, 208, 285, 307, 260, 175]) #横坐标
y_data = np.array([205, 124, 135, 173, 216, 311, 309])-100 #纵坐标
pts =np.c_[x_data, y_data] #合并横纵坐标
# print(pts)
cv2.fillPoly(mask,[pts], (255), 8, 0)
cv2.imshow("mask",mask)
cv2.waitKey(0)
cv2.destroyAllWindows()
import cv2
img = cv2.imread(r"C:\Users\Curry\Desktop\flower.png")
hsv = cv2.cvtColor(img, code=cv2.COLOR_BGR2HSV)
h,w,c = img.shape
mask = np.zeros((h, w), dtype=np.uint8)
x_data = np.array([124, 169, 208, 285, 307, 260, 175]) #横坐标
y_data = np.array([205, 124, 135, 173, 216, 311, 309])-100 #纵坐标
pts =np.c_[x_data, y_data] #合并横纵坐标
# print(pts)
cv2.fillPoly(mask,[pts], (255), 8, 0)
res = cv2.bitwise_and(img,img,mask=mask)
cv2.imshow("mask", res)
cv2.waitKey(0)
cv2.destroyAllWindows()