Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import cv2
- import numpy as np
- RED = [
- [160,100,100],
- [179,255,255]
- ]
- BLUE = [
- [110,50,50],
- [130, 255,255]
- ]
- def isolate_by_color(filename="blueball.jpg", lower_bound=[110,50,50], upper_bound=[130, 255,255], color=None):
- img = cv2.imread(filename)
- hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
- if color == None:
- lower_color = np.array(lower_bound)
- upper_color = np.array(upper_bound)
- mask = cv2.inRange(hsv, lower_color, upper_color)
- else:
- lower_color1 = np.array(color[0])
- upper_color1 = np.array(color[1])
- mask = cv2.inRange(hsv, lower_color1, upper_color1)
- #mask = cv2.inRange(mask, lower_color1, upper_color1)
- res = cv2.bitwise_and(img, img, mask=mask)
- cv2.imshow("result",res)
- k = cv2.waitKey(0)
- if k == ord('q'):
- cv2.destroyAllWindows()
- if __name__ == "__main__":
- #isolate_by_color("blueball.jpg")
- #isolate_by_color("girl.jpeg", lower_bound=[0, 100, 100], upper_bound=[10,255,255])
- isolate_by_color("girl.jpeg", color=red)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement