Advertisement
Guest User

Untitled

a guest
Aug 18th, 2019
68
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.72 KB | None | 0 0
  1. import cv2
  2.  
  3. car_cascade = cv2.CascadeClassifier('cars.xml')
  4. cap =cv2.VideoCapture('car3.mp4')
  5.  
  6. frame_width = int(cap.get(3))
  7. frame_height = int(cap.get(4))
  8.  
  9. out = cv2.VideoWriter('output.avi',cv2.VideoWriter_fourcc('M','J','P','G'), 10, (frame_width,frame_height))
  10.  
  11. while True:
  12. ret, frame = cap.read()
  13. if ret:
  14. gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
  15. cars = car_cascade.detectMultiScale(gray, 1.3, 5)
  16. for(x, y, w, h) in cars:
  17. cv2.rectangle(frame, (x,y), (x+w, y+h), (255,0,0), 2)
  18.  
  19. cv2.imshow('Car Detection', frame)
  20. out.write(frame)
  21. if cv2.waitKey(1) & 0xFF == ord('q'):
  22. break
  23. else:
  24. break
  25.  
  26. cap.release()
  27. out.release()
  28. cv2.destroyAllWindows()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement