Update app.py

app.py

@@ -164,8 +164,8 @@ def removeSmallDashes(imgOriginal,green):
   img1=cv2.dilate(smalldashes, kernel3, iterations=2)
   img2=cv2.erode(img1, kernel3, iterations=2)
 
-  smalldashes=cv2.medianBlur(img2,7)
-  smalldashes=cv2.medianBlur(smalldashes,7)
+  smalldashes=cv2.medianBlur(img2,5)
+  smalldashes=cv2.medianBlur(smalldashes,5)
   # cv2_imshow(smalldashes)
   smalldashesOut=green.copy()
   smalldashesOut=cv2.cvtColor(smalldashesOut,cv2.COLOR_GRAY2BGR)
@@ -287,29 +287,35 @@ def hexRGB(color):
   color= tuple(int(color[i:i+2], 16) for i in (0, 2, 4)) #hex to rgb
   color=np.array(color) #rgb to bgr 
   return color
-
 def DetectColor(img,color=0):
 
   imgCopy=img.copy()
-  tol=25 #tolerance
-  #rgb values
+  imgCopy=cv2.cvtColor(imgCopy,cv2.COLOR_BGR2HSV)
+  tol=5 #tolerance
   color=hexRGB(color) 
+  h,s,v = cv2.cvtColor(np.uint8([[[color[2],color[1],color[0]]]]),cv2.COLOR_BGR2HSV)[0][0]
+
+  lower =np.array( [h- tol, 100, 100 ], dtype='uint8')
+  upper = np.array( [h + tol, 255, 255],dtype='uint8')
 
-  lower = np.array([max(color[2]-tol,0), max(color[1]-tol,0), max(color[0]-tol,0)], dtype = "uint8")  #BGR
-  upper= np.array([min(color[2]+tol,255), min(color[1]+tol,255), min(color[0]+tol,255)], dtype = "uint8")
-  mask = cv2.inRange(imgCopy, lower, upper)
+  mask = cv2.inRange(imgCopy, lower , upper)
 
   detectedColors = cv2.bitwise_and(imgCopy,imgCopy, mask= mask)   # Bitwise-AND mask and original image
   
   kernel=np.ones((3,3),np.uint8)
   mask=cv2.dilate(mask,kernel, iterations=5)
   mask=cv2.erode(mask,kernel, iterations=4)
-  # cv2_imshow(mask)
 
   detectedColors=cv2.dilate(detectedColors,kernel, iterations=5)
   detectedColors=cv2.erode(detectedColors,kernel, iterations=4)
+
+  detectedColors=cv2.cvtColor(detectedColors,cv2.COLOR_HSV2BGR)
+  detectedColors=cv2.medianBlur(detectedColors,7)
+  # cv2_imshow(detectedColors)
+  
   return mask, detectedColors, color
 
+    
 def detectAllColors(img,finalColorArray):
   for i in range(len(finalColorArray)):
     detectedColors= DetectColor(img,finalColorArray[i])[1]