diff --git a/data/LegosBlue/blue1.JPG b/data/LegosBlue/blue1.JPG new file mode 100644 index 0000000..7427530 Binary files /dev/null and b/data/LegosBlue/blue1.JPG differ diff --git a/data/LegosBlue/blue2.JPG b/data/LegosBlue/blue2.JPG new file mode 100644 index 0000000..34ea344 Binary files /dev/null and b/data/LegosBlue/blue2.JPG differ diff --git a/data/LegosBlue/blue3.JPG b/data/LegosBlue/blue3.JPG new file mode 100644 index 0000000..0dd8b39 Binary files /dev/null and b/data/LegosBlue/blue3.JPG differ diff --git a/data/LegosBlue/blue4.JPG b/data/LegosBlue/blue4.JPG new file mode 100644 index 0000000..95f266b Binary files /dev/null and b/data/LegosBlue/blue4.JPG differ diff --git a/data/LegosBlue/blue5.JPG b/data/LegosBlue/blue5.JPG new file mode 100644 index 0000000..332ccb9 Binary files /dev/null and b/data/LegosBlue/blue5.JPG differ diff --git a/watershed.py b/watershed.py index da3b140..10d4bb8 100644 --- a/watershed.py +++ b/watershed.py @@ -5,9 +5,14 @@ import os print('-----') dirname = os.path.dirname(__file__) -filename = os.path.join(dirname,'data/legos3.jpg') +filename = os.path.join(dirname,'data/LegosBlue/blue5.jpg') print(os.path.exists(filename)) img = cv2.imread(filename) +hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV) +lwr = np.array([85,50,40]) +upr = np.array([135,255,255]) +msk = cv2.inRange(hsv, lwr, upr) +img[msk>0]=(0,0,0) b,g,r = cv2.split(img) rgb_img = cv2.merge([r,g,b])