Merge pull request #243 from kritikaparmar-programmer/color

Added script for Color Detection #204

Author: powerexploit

Image-Processing/Color_Detection/Readme.md

@@ -0,0 +1,33 @@
+# Color Detection
+
+This is a python script which we can use to determine the color of a part of image.
+
+## Installation
+
+* Use the package manager [pip](https://pip.pypa.io/en/stable/) to install Opencv.
+
+```bash
+pip install opencv-python
+``` 
+* Install Numpy
+
+```bash
+pip install numpy
+```
+* Install Pandas
+
+```bash
+pip install pandas
+```
+
+## How it works
+
+* When the user double clicks on the image, then color of that part of image is shown in a box.
+
+## How to run in CMD
+
+> python color_detection.py -i <colorpic.jpg>
+
+## Screenshot
+
+![](../Color_Detection/color_sample.png)

Image-Processing/Color_Detection/color_detection.py

@@ -0,0 +1,75 @@
+import cv2
+import numpy as np
+import pandas as pd
+import argparse
+
+# Using argparse library can directly give an image path from the command prompt
+ap = argparse.ArgumentParser()
+ap.add_argument('-i', '--image', required=True, help="Image Path")
+args = vars(ap.parse_args())
+img_path = args['image']
+
+# Reading image with opencv
+img = cv2.imread(img_path)
+
+# Global Variables
+clicked = False
+r = g = b = xpos = ypos = 0
+
+
+# Reading csv file using pandas
+index = ["color", "color_name", "hex", "R", "G", "B"]
+csv = pd.read_csv('colors.csv', names=index, header=None)
+
+
+def draw_function(event, x, y, flags, param):
+    if event == cv2.EVENT_LBUTTONDBLCLK:
+        global b, g, r, xpos, ypos, clicked
+        clicked = True
+        xpos = x
+        ypos = y
+        b, g, r = img[y, x]
+        b = int(b)
+        g = int(g)
+        r = int(r)
+      
+
+def getColorName(R, G, B):
+    minimum = 10000
+    for i in range(len(csv)):
+        d = abs(R - int(csv.loc[i, "R"])) + abs(G - int(csv.loc[i, "G"])) + abs(B - int(csv.loc[i, "B"]))
+        if(d<=minimum):
+            minimum = d
+            cname = csv.loc[i, "color_name"]
+    return cname
+
+
+cv2.namedWindow('image')
+cv2.setMouseCallback('image',draw_function)
+
+
+while(1):
+
+    cv2.imshow("image",img)
+    if (clicked):
+   
+        # cv2.rectangle(image, startpoint, endpoint, color, thickness)-1 fills entire rectangle 
+        cv2.rectangle(img,(20,20), (750,60), (b,g,r), -1)
+
+        # Creating text string to display( Color name and RGB values )
+        text = getColorName(r,g,b) + ' R='+ str(r) +  ' G='+ str(g) +  ' B='+ str(b)
+        
+        # cv2.putText(img,text,start,font(0-7),fontScale,color,thickness,lineType )
+        cv2.putText(img, text,(50,50),2,0.8,(255,255,255),2,cv2.LINE_AA)
+
+        # For very light colours display text in black colour
+        if(r+g+b>=600):
+            cv2.putText(img, text,(50,50),2,0.8,(0,0,0),2,cv2.LINE_AA)
+            
+        clicked=False
+
+    # Break the loop when user hits 'esc' key    
+    if cv2.waitKey(20) & 0xFF ==27:
+        break
+    
+cv2.destroyAllWindows()