Advertisement
Andrerb

Captura imagens

May 31st, 2018
370
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.74 KB | None | 0 0
  1. import cv2
  2. import numpy as np
  3.  
  4. classificador = cv2.CascadeClassifier("haarcascade_frontalface_default.xml")
  5. classificadorOlho = cv2.CascadeClassifier("haarcascade_eye.xml")
  6.  
  7.  
  8. camera = cv2.VideoCapture(0)
  9. amostra=1
  10. numeroAmostra=20
  11. id=input('Digite seu identificador: ')
  12. largura,altura = 220 ,220
  13.  
  14. print("capture ...")
  15.  
  16. while (True):
  17.     conectado,imagem = camera.read()
  18.     imagemCinza = cv2.cvtColor(imagem, cv2.COLOR_BGR2GRAY)
  19.     print(np.average(imagemCinza))
  20.     facesDetectadas = classificador.detectMultiScale(imagemCinza,
  21.                                                      scaleFactor=1.5,
  22.                                                      minSize=(150,150))
  23.    
  24.    
  25.     for(x,y,l,a) in facesDetectadas:
  26.        
  27.         cv2.rectangle(imagem, (x,y),(x+l,y+a), (0,0,255),2)
  28.         regiao = imagem[y:y + a , x:x +l]
  29.         regiaoCinzaOlho = cv2.cvtColor(regiao,cv2.COLOR_BGR2GRAY)
  30.         olhosDetectado = classificador.detectMultiScale(regiaoCinzaOlho)
  31.        
  32.         for(ox. oy, ol, oa) in olhosDetectado:
  33.             cv2.rectangle(regiao, (ox,oy), (ox + ol , oy + oa), (0,255,0),2)
  34.            
  35.             if cv2.waitKey(1)& 0xFF == ord('q'):
  36.                 if np.average(imagemCinza) > 100:
  37.                     imagemFace = cv2.resize(imagemCinza[y:y+a,x:x +l], (largura,altura))
  38.                     cv2.imwrite("fotos1/pessoa. " + str(id) + "." +str(amostra) + ".jpg", imagemFace)
  39.                     print("[ foto " + str(amostra) + " capturada com sucesso ]" )
  40.                     amostra +=1
  41.            
  42.          
  43.     cv2.imshow("Face",imagem)
  44.     cv2.waitKey(1)
  45.    
  46.     if(amostra >= numeroAmostra +1 ):
  47.         break
  48.    
  49.  
  50.  
  51. print("faces detectada com sucesso")
  52. camera.release()
  53. cv2.destroyAllWindows()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement