From 6abd03e1bedb0bc1866baba77e5f1e0dff080300 Mon Sep 17 00:00:00 2001
From: Divakar Lakhera <divakarlakhera99@gmail.com>
Date: Tue, 8 Oct 2019 02:07:05 +0530
Subject: [PATCH] Added new functionality

Added code for green, blue and red colour detection and added code to draw borders around them.
---
 color_detect.py | 104 +++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 86 insertions(+), 18 deletions(-)

diff --git a/color_detect.py b/color_detect.py
index e31e3aa..a12360b 100755
--- a/color_detect.py
+++ b/color_detect.py
@@ -1,31 +1,99 @@
 #!/usr/bin/env python
+
 import numpy as np
 import cv2
 
+# Set Capture Device as Webcam
 cap = cv2.VideoCapture(0)
+"""
+        def greenCircleDetect(frame)
+            Returns frame with contour drawn around all Green Objects.
+            Requires fr(frame) to work on.
+"""
+def greenCircleDetect(fr):
+    # UpperBound and LowerBound HSV Range for GREEN colour.
+    lowerBound = np.array([33, 80, 40])
+    upperBound = np.array([102, 255, 255])
+    # First Convert Frame to HSV
+    imgHSV = cv2.cvtColor(fr, cv2.COLOR_BGR2HSV)
+    # Now Create a Mask for GREEN Colour
+    mask = cv2.inRange(imgHSV, lowerBound, upperBound)
+    # Now BitwiseAND HSV and mask
+    finalFrame = cv2.bitwise_and(imgHSV, imgHSV, mask=mask);
+    # Convert final frame to grayscale before contour detection.
+    gray_image = cv2.cvtColor(finalFrame, cv2.COLOR_BGR2GRAY)
+    # create a thresh image out of grayscale image provided.
+    ret, thresh = cv2.threshold(gray_image, 127, 255, 0)
+    # detect contours and draw border around it with black(0,0,0) colour.
+    ctrs, hrc = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
+    cv2.drawContours(fr, ctrs, -1, (0, 0, 0), 2)
+    return fr
 
-def greenCircleDetect():
-
-
-def redCircleDetect():
+"""
+        def redCircleDetect(frame)
+            Returns frame with contour drawn around all Red Objects.
+            Requires fr(frame) to work on.
+"""
+def redCircleDetect(fr):
+    # UpperBound and LowerBound HSV Range for Red colour for Lower Saturation Range.
+    lower_red = np.array([0, 50, 50])
+    upper_red = np.array([10, 255, 255])
+    # First Convert Frame to HSV
+    imgHSV = cv2.cvtColor(fr, cv2.COLOR_BGR2HSV)
+    # Now Create a Mask for Low Saturation Blue colour.
+    mask0 = cv2.inRange(imgHSV, lower_red, upper_red)
+    # UpperBound and LowerBound HSV Range for Red colour for Higher Saturation Range.
+    lower_red = np.array([170, 50, 50])
+    upper_red = np.array([180, 255, 255])
+    # Now Create a Mask for High Saturation Blue colour.
+    mask1 = cv2.inRange(imgHSV, lower_red, upper_red)
+    #add Both the masks to make complete mask covering both high and low saturation ranges
+    mask = mask1 + mask0;
+    # Mask it Out
+    finalFrame = cv2.bitwise_and(imgHSV, imgHSV, mask=mask);
+    # GreyScale Conversion
+    gray_image = cv2.cvtColor(finalFrame, cv2.COLOR_BGR2GRAY)
+    # threshold conversion
+    ret, thresh = cv2.threshold(gray_image, 127, 255, 0)
+    # contour detection
+    ctrs, hrc = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
+    cv2.drawContours(fr, ctrs, -1, (0, 0, 0), 2)
+    return fr
 
+"""
+        def blueCircleDetect(frame)
+            Returns frame with contour drawn around all Blue Objects.
+            Requires fr(frame) to work on.
+"""
 
-def blueCircleDetect():
+def blueCircleDetect(fr):
+    # UpperBound and LowerBound HSV Range for Blue colour.
+    lowerBound = np.array([100, 150, 0])
+    upperBound = np.array([140, 255, 255])
+    # First Convert Frame to HSV
+    imgHSV = cv2.cvtColor(fr, cv2.COLOR_BGR2HSV)
+    # Now Create a Mask for GREEN Colour
+    mask = cv2.inRange(imgHSV, lowerBound, upperBound)
+    # Now BitwiseAND HSV and mask
+    finalFrame = cv2.bitwise_and(imgHSV, imgHSV, mask=mask);
+    # Convert final frame to grayscale before contour detection.
+    gray_image = cv2.cvtColor(finalFrame, cv2.COLOR_BGR2GRAY)
+    # create a thresh image out of grayscale image provided.
+    ret, thresh = cv2.threshold(gray_image, 127, 255, 0)
+    # detect contours and draw border around it with black(0,0,0) colour.
+    ctrs, hrc = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
+    cv2.drawContours(fr, ctrs, -1, (0, 0, 0), 2)
+    return fr
 
 
 while True:
-	ret, frame = cap.read()
-
-	greenCircleDetect()
-
-	redCircleDetect()
-
-	blueCircleDetect()
-
-
-	cv2.imshow('frame', frame)
-
-	cv2.waitKey(3)
+    # Get a frame from Webcam.
+    ret, frame = cap.read()
+    # Process the frame for Green-->Red-->Blue colours and draw border around them.
+    frame=blueCircleDetect(redCircleDetect(greenCircleDetect(frame)))
+    #Display the final frame
+    cv2.imshow('frame', frame)
+    cv2.waitKey(3)
 
 cap.release()
-cv2.destroyAllWindows()
\ No newline at end of file
+cv2.destroyAllWindows()