Spaces:
No application file
No application file
eddygiusepe
commited on
Commit
·
ad5dfc9
1
Parent(s):
564e80a
opencv com vídeo
Browse files
3_Salvando_em_video_ReconhecimentoFacial.py
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import cv2
|
| 2 |
+
import mediapipe as mp
|
| 3 |
+
|
| 4 |
+
webcam = cv2.VideoCapture(0) # Para conectar o Python com a webcam.
|
| 5 |
+
reconhecimento_rosto = mp.solutions.face_detection # Ativando a solução de reconhecimento de rosto
|
| 6 |
+
desenho = mp.solutions.drawing_utils # Ativando a solução de desenho
|
| 7 |
+
reconhecedor_rosto = reconhecimento_rosto.FaceDetection() # Criando o item que consegue ler uma imagem e reconhecer os rostos ali dentro
|
| 8 |
+
|
| 9 |
+
# Configuração para salvar o vídeo
|
| 10 |
+
fourcc = cv2.VideoWriter_fourcc(*'XVID')
|
| 11 |
+
out = cv2.VideoWriter('detecao_face.avi', fourcc, 20.0, (640, 480))
|
| 12 |
+
|
| 13 |
+
while webcam.isOpened():
|
| 14 |
+
validacao, frame = webcam.read() # Lê a imagem da webcam
|
| 15 |
+
if not validacao:
|
| 16 |
+
break
|
| 17 |
+
imagem = frame
|
| 18 |
+
lista_rostos = reconhecedor_rosto.process(imagem) # usa o reconhecedor para criar uma lista com os rostos reconhecidos
|
| 19 |
+
|
| 20 |
+
if lista_rostos.detections: # caso algum rosto tenha sido reconhecido
|
| 21 |
+
for rosto in lista_rostos.detections: # para cada rosto que foi reconhecido
|
| 22 |
+
desenho.draw_detection(imagem, rosto) # desenha o rosto na imagem
|
| 23 |
+
out.write(imagem) # Salva o quadro com a detecção no arquivo de vídeo
|
| 24 |
+
|
| 25 |
+
cv2.imshow("Rostos na sua webcam", imagem) # mostra a imagem da webcam para você
|
| 26 |
+
if cv2.waitKey(5) == 27: # ESC # garante que o código vai ser pausado ao apertar ESC (código 27) e que o código vai esperar 5 milissegundos a cada leitura da webcam
|
| 27 |
+
break
|
| 28 |
+
|
| 29 |
+
webcam.release() # encerra a conexão com a webcam
|
| 30 |
+
out.release() # fecha o arquivo de vídeo
|
| 31 |
+
cv2.destroyAllWindows() # fecha a janela que mostra o que a webcam está vendo
|