colourdetection.py

# -*- coding: utf-8 -*-
"""
Created on Sun Mar  1 00:07:29 2020

@author: MARIUM ALI
"""

import cv2
import pandas as pd

   

    
img_path="C:\\Users\\MARIUM ALI\\Desktop\\python-project-color-detection (1)\\colorpic.jpg"
#reading image using opencv
img=cv2.imread(img_path)

#initializing global var
clicked=False
r=g=b=xpos=ypos=0

#displaying image using opencv
    #cv2.imshow('image',img)
    #cv2.waitKey(0)
    #cv2.destroyAllWindows()

index=["colors","name","hex","R","G","B"]
csv=pd.read_csv('C:\\Users\\MARIUM ALI\\Desktop\\python-project-color-detection (1)\\colors.csv',names=index,header=None)

def getcolor(R,G,B):
    mini=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<=mini):
            mini=d
            cname=csv.loc[i,"name"]
    return cname


def draw_function(event,x,y,flags,param):
    if (event==cv2.EVENT_LBUTTONDBLCLK):
        global b,g,r,clicked,xpos,ypos
        clicked=True
        xpos=x
        ypos=y
        b,g,r=img[y,x]
        b=int(b)
        g=int(g)
        r=int(r)

#name image window and respond to click on the window
cv2.namedWindow('image')
cv2.setMouseCallback('image',draw_function)  #draw_func is a func that will identify the color

while(1):
    cv2.imshow("image",img)
    if(clicked):
        cv2.rectangle(img,(20,20),(750,60),(b,g,r),-1)
        text=getcolor(r,g,b) + ' R='+str(r)+' G='+str(g)+' B='+str(b)
        cv2.putText(img,text,(50,50),2,0.8,(255,255,255),2,cv2.LINE_AA)
        
        if(r+g+b>=600):
            cv2.putText(img, text,(50,50),2,0.8,(0,0,0),2,cv2.LINE_AA)
        clicked=False
    if(cv2.waitKey(20) & 0xFF==27):
        break
cv2.destroyAllWindows()