import cv2
import numpy as np
img = cv2.imread('RED_2thr.png')
img = cv2.imread('red_4.png')
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
lower_red = np.array([0, 50, 50])
upper_red = np.array([10, 255, 255])
mask1 = cv2.inRange(hsv, lower_red, upper_red)
lower_red = np.array([170, 50, 50])
upper_red = np.array([180, 255, 255])
mask2 = cv2.inRange(hsv, lower_red, upper_red)
red_1 = cv2.bitwise_and(img, img, mask=mask1)
red_2 = cv2.bitwise_and(img, img, mask=mask2)
mask = mask1 + mask2
red_only = cv2.bitwise_and(img, img, mask=mask)
#cv2.imshow('Original Image', img)
#cv2.imshow('Red Only', red_only)
vstk1=np.vstack((img,red_only,red_1,red_2))
vstk2=np.vstack((mask1,mask2))
cv2.imshow('vstk2',vstk2)
cv2.imshow('vstk1',vstk1)
cv2.waitKey(0)
cv2.destroyAllWindows()