结果:
import cv2
import numpy as np
imgs = cv2.imread("timg.jpg")
image = imgs.copy()
images = imgs.copy()
print(image.shape)
rows, cols = image.shape[:2]
print(rows, cols)
red_minus_blue = image[:, :, 2] - image[:, :, 0]
red_minus_green = image[:, :, 2] - image[:, :, 1]
red_minus_blue = red_minus_blue >= 20
red_minus_green = red_minus_green >= 20
red = image[:, :, 2] >= np.mean(image[:, :, 2])/2
mask = red_minus_green & red_minus_blue & red
print(mask)
images[mask,:]=255
mask = (1-mask).astype(np.bool)
print(mask)
image[mask, :] = 255
stack = np.vstack([imgs,image,images])
cv2.imshow("stack", stack)
cv2.imwrite("result.jpg",stack)
# cv2.imshow("orgin", imgs)
# cv2.imshow("red", image)
# cv2.imshow("delete_red", images)
cv2.waitKey()
# 红色像素值最大,且大于阈值(中值)
# 其他通道像素值的距离比较小,且与红色像素值的距离比率较大,且大于阈值(自己调)。