spacepaste

  1.  
  2. import os
  3. import numpy as np
  4. import cv2
  5. path = '/home/agent/PythonOpen/cvpics'
  6. face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
  7. eye_cascade = cv2.CascadeClassifier('haarcascade_eye.xml')
  8. cap = cv2.VideoCapture('testvid.mp4')
  9. num = 0
  10. while 1:
  11. ret, img = cap.read()
  12. gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
  13. faces = face_cascade.detectMultiScale(gray, 1.3, 5)
  14. for (x,y,w,h) in faces:
  15. cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2), cv2.imwrite(os.path.join(path, 'opencv'+str(num)+'.jpg',img))
  16. roi_gray = gray[y:y+h, x:x+w]
  17. roi_color = img[y:y+h, x:x+w]
  18. eyes = eye_cascade.detectMultiScale(roi_gray)
  19. for (ex,ey,ew,eh) in eyes:
  20. cv2.rectangle(roi_color,(ex,ey),(ex+ew,ey+eh),(0,255,0),2)
  21. cv2.imshow('img',img)
  22. num = num+1
  23. k = cv2.waitKey(30) & 0xff
  24. if k == 27:
  25. break
  26. cap.release()
  27. cv2.destroyAllWindows()
  28.