去除圖像中的紅色部分（or 保留） opencv-python

結果：

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

# 紅色像素值最大，且大於閾值（中值）
# 其他通道像素值的距離比較小，且與紅色像素值的距離比率較大，且大於閾值（自己調）。