Detector_Cores_GY_Py_OpCV.py

#Programa com Python e OpenCV que identifica e mostra cores em imagens e vídeos.

#Ao rodar, o programa abre a webcam e consegue identificar as cores verde e amarelo, 
# pois são as únicas máscaras que estão configuradas até o momento.

#Importando as bibliotecas

import cv2
import argparse  # O módulo para fazer a Análise de Linha de Comando 
from operator import xor # Operação XOR bit a bit 

# As função def - irá agrupar um conjunto de instruções em um bloco, 
# permitindo que esse bloco seja executado quantas vezes forem necessárias.

def callback(value):
    pass

# Função setup_trackbar(), o primeiro argumento é o nome do trackbar, 
# o segundo é o nome da janela à qual está anexado, 
# o terceiro argumento é o valor padrão, o quarto é o valor máximo 
# e o quinto é a função de retorno de chamada que é executada 
# toda vez que o trackbar alterações de valor.



def setup_trackbars(range_filter):
    cv2.namedWindow("Trackbars", 0)

    for i in ["MIN", "MAX"]:  //loop para os valores máximos e mínimos 
        v = 0 if i == "MIN" else 255

        for j in range_filter:
            cv2.createTrackbar("%s_%s" % (j, i), "Trackbars", v, 255, callback)

# A função que chama os argumentos 
# Analisador sintático para opções de linha de comando, argumentos e subcomandos

def get_arguments():
    ap = argparse.ArgumentParser()
    ap.add_argument('-f', '--filter', required=True,
                    help='Range filter. RGB or HSV')
    ap.add_argument('-i', '--image', required=False,
                    help='Path to the image')
    ap.add_argument('-w', '--webcam', required=False,
                    help='Use webcam', action='store_true')
    ap.add_argument('-p', '--preview', required=False,
                    help='Show a preview of the image after applying the mask',
                    action='store_true')
    args = vars(ap.parse_args())

    if not xor(bool(args['image']), bool(args['webcam'])):
        ap.error("Please specify only one image source")

    if not args['filter'].upper() in ['RGB', 'HSV']:
        ap.error("Please speciy a correct filter.")

    return args


def get_trackbar_values(range_filter):
    values = []

    for i in ["MIN", "MAX"]:
        for j in range_filter:
            v = cv2.getTrackbarPos("%s_%s" % (j, i), "Trackbars")
            values.append(v)

    return values

# O ponto de entrada padrão do programa

def main():
    args = get_arguments()

    range_filter = args['filter'].upper()

    if args['image']:
        image = cv2.imread(args['image'])

        if range_filter == 'RGB':
            frame_to_thresh = image.copy()
        else:
            frame_to_thresh = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
    else:
        camera = cv2.VideoCapture(0)

    setup_trackbars(range_filter)

    while True:
        if args['webcam']:
            ret, image = camera.read()

            if not ret:
                break

            if range_filter == 'RGB':
                frame_to_thresh = image.copy()
            else:
                frame_to_thresh = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)

        v1_min, v2_min, v3_min, v1_max, v2_max, v3_max = get_trackbar_values(range_filter)

        thresh = cv2.inRange(frame_to_thresh, (v1_min, v2_min, v3_min), (v1_max, v2_max, v3_max))

        if args['preview']:
            preview = cv2.bitwise_and(image, image, mask=thresh)
            cv2.imshow("Preview", preview)
        else:
            cv2.imshow("Original", image)
            cv2.imshow("Thresh", thresh)

        if cv2.waitKey(1) & 0xFF is ord('q'):
            break


if __name__ == '__main__':
    main()