You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

27 lines
606 B

  1. from gridEyeReader import GridEye
  2. from dataFuser import DataFuser
  3. import cv2
  4. import logging
  5. import time
  6. import numpy as np
  7. logging.basicConfig(level=logging.DEBUG)
  8. g = GridEye('/dev/ttyACM0')
  9. d = DataFuser(10) # distance between sensors.
  10. cv2.namedWindow("Image", cv2.WINDOW_NORMAL)
  11. for _ in range(1000000):
  12. frame = g.readData()
  13. out = d.mergeFrame(np.array(frame), 100)
  14. if type(out) is np.ndarray:
  15. cv2.imshow('Image', cv2.cvtColor(out, cv2.COLOR_GRAY2BGR))
  16. k = cv2.waitKey(1)
  17. if k == ord('q'):
  18. break
  19. elif k == ord('c'):
  20. cv2.imwrite('shot.png', out)
  21. cv2.destroyAllWindows()