-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcapturando 1.py
64 lines (51 loc) · 1.63 KB
/
capturando 1.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import cv2
import os
# Cria as pastas se não existirem
folders = ['a', 'w', 'd']
for folder in folders:
if not os.path.exists(folder):
os.makedirs(folder)
# Abre a webcam (geralmente o índice 0 se refere à primeira webcam USB conectada)
cap = cv2.VideoCapture(0)
# Verifica se a webcam foi aberta corretamente
if not cap.isOpened():
print("Erro ao abrir a webcam")
exit()
# Inicializa contadores de imagens para cada pasta
counters = {folder: 0 for folder in folders}
while True:
# Captura um frame da webcam
ret, frame = cap.read()
# Verifica se o frame foi capturado corretamente
if not ret:
print("Erro ao capturar o frame")
break
# Mostra o frame em uma janela
cv2.imshow('Webcam', frame)
# Verifica se uma tecla foi pressionada
key = cv2.waitKey(1) & 0xFF
if key == ord('q'):
# Se a tecla 'q' for pressionada, sai do loop
break
elif key == ord('a'):
# Salva a imagem na pasta 'a'
filename = f'a/{counters["a"]}.jpg'
cv2.imwrite(filename, frame)
counters['a'] += 1
print(f'Imagem {filename} salva')
elif key == ord('w'):
# Salva a imagem na pasta 'w'
filename = f'w/{counters["w"]}.jpg'
cv2.imwrite(filename, frame)
counters['w'] += 1
print(f'Imagem {filename} salva')
elif key == ord('d'):
# Salva a imagem na pasta 'd'
filename = f'd/{counters["d"]}.jpg'
cv2.imwrite(filename, frame)
counters['d'] += 1
print(f'Imagem {filename} salva')
# Libera a webcam
cap.release()
# Fecha todas as janelas abertas
cv2.destroyAllWindows()