Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import numpy as np
- import cv2
- import os
- import sys
- #validator parametru imienia
- your_name= sys.argv[1:]
- if len(your_name) == 0:
- print("type your name-surname as a parameter")
- exit()
- os.makedirs('images/'+str(your_name))
- face_cascade = cv2.CascadeClassifier('cascades/data/haarcascade_frontalface_alt2.xml')
- cap = cv2.VideoCapture(0)
- i = 0
- while(True):
- #przechwytywanie obrazu
- ret, frame = cap.read()
- # operacje na poszczególnych klatkach (zmiana koloru na czarno-biały etc)
- gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
- faces = face_cascade.detectMultiScale(gray, scaleFactor=1.5, minNeighbors=5)
- for (x, y, w, h) in faces:
- #print(x,y,w,h)
- roi_gray = gray[y:y+h, x:x+w]
- roi_color = frame[y:y+h, x:x+w]
- #zapisywanie klatek do folderu
- i=i+1
- print(i)
- cv2.imwrite('images/'+str(your_name)+'/picture'+str(i)+'.png', frame)
- #po 200 zdjęciach kończy przechwytywanie obrazu
- if i == 200: break
- #wyświetlenie obrazu oraz zamknięcie okna
- cv2.imshow('frame',frame)
- cv2.imshow('gray',gray)
- if cv2.waitKey(20) & 0xFF == ord('q'):
- break
- cap.release()
- cv2.destroyAllWindows()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement