48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
import cv2
|
|
import numpy as np
|
|
|
|
def detect_green(image):
|
|
# Converter a imagem de BGR para HSV
|
|
hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
|
|
|
|
# Definir os intervalos de cor verde (pode precisar ajustar os valores)
|
|
lower_green = np.array([35, 50, 50])
|
|
upper_green = np.array([90, 255, 255])
|
|
|
|
# Criar uma máscara para a cor verde na faixa especificada
|
|
mask = cv2.inRange(hsv, lower_green, upper_green)
|
|
|
|
# Encontrar os contornos na máscara
|
|
contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
|
|
|
|
# Criar uma imagem preta para desenhar as caixas delimitadoras
|
|
black = np.zeros_like(image)
|
|
|
|
# Desenhar caixas delimitadoras ao redor dos contornos detectados na imagem preta
|
|
for contour in contours:
|
|
x, y, w, h = cv2.boundingRect(contour)
|
|
cv2.rectangle(black, (x, y), (x + w, y + h), (0, 255, 0), 2)
|
|
|
|
# Mostrar apenas as áreas verdes na imagem original
|
|
result = cv2.bitwise_and(image, image, mask=mask)
|
|
result = cv2.add(result, black) # Adicionar as caixas delimitadoras à imagem original
|
|
|
|
return result
|
|
|
|
cap = cv2.VideoCapture(1)
|
|
|
|
while True:
|
|
ret, frame = cap.read()
|
|
if not ret:
|
|
break
|
|
|
|
# Detectar áreas verdes na imagem e mostrar apenas elas com as caixas delimitadoras
|
|
green_boxes = detect_green(frame)
|
|
|
|
cv2.imshow('Green Detection', green_boxes)
|
|
if cv2.waitKey(1) & 0xFF == ord('q'):
|
|
break
|
|
|
|
cap.release()
|
|
cv2.destroyAllWindows()
|