%pylab inline
import cv2

def myShow(im_title, im):
    ''' This is function to display the image'''
    plt.figure()  
    plt.title(im_title)
    plt.axis("off")
    if len(im.shape) == 2:
        plt.imshow(im, cmap = "gray")
    else:
        im_display = cv2.cvtColor(im, cv2.COLOR_RGB2BGR)
        plt.imshow(im_display)
    plt.show()

im = cv2.imread("manu.jpg")
myShow("Input Image", im)

im_red_ball_mask = cv2.inRange(im, (0, 0, 30), (80, 80, 255))
im_red_ball_mask = im_red_ball_mask.astype('bool')
myShow("Resulting Image", im * np.dstack((im_red_ball_mask, im_red_ball_mask, im_red_ball_mask)))

im_hsv = cv2.cvtColor(im, cv2.COLOR_BGR2HSV)

im_red_ball_mask_1 = cv2.inRange(im_hsv, (0, 100, 0), (10, 255, 255))
im_red_ball_mask_1 = im_red_ball_mask_1.astype('bool')
myShow("Resulting image with Hue values within 0-10", im * np.dstack((im_red_ball_mask_1, im_red_ball_mask_1, im_red_ball_mask_1)))

im_red_ball_mask_2 = cv2.inRange(im_hsv, (170, 100, 0), (180, 255, 255)) 
im_red_ball_mask_2 = im_red_ball_mask_2.astype('bool')
myShow("Resulting image with Hue values within 170-180", im * np.dstack((im_red_ball_mask_2, im_red_ball_mask_2, im_red_ball_mask_2)))

im_red_ball_mask_full = im_red_ball_mask_1 + im_red_ball_mask_2
myShow("Resulting image by adding both the masks", im * np.dstack((im_red_ball_mask_full, im_red_ball_mask_full, im_red_ball_mask_full)))