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.
46 lines
1.8 KiB
46 lines
1.8 KiB
import cv2
|
|
import numpy as np
|
|
import struct
|
|
import time
|
|
import sys
|
|
|
|
log_file = sys.argv[1] if len(sys.argv) > 1 else 'lepton_log.bin'
|
|
output_folder = 'videos'
|
|
|
|
fourcc = cv2.VideoWriter_fourcc(*'XVID')
|
|
leptonVideoWriter = cv2.VideoWriter(output_folder+'/lepton.avi', fourcc, 10.0, (160, 120), False)
|
|
gridEyeVideoWriter = cv2.VideoWriter(output_folder+'/gridEye.avi', fourcc, 10.0, (320, 320), False)
|
|
#cv2.resizeWindow('image', 320, 320)
|
|
with open(log_file, 'rb') as f:
|
|
while 1:
|
|
leptonData = f.read(38400)
|
|
if len(leptonData) != 38400:
|
|
break
|
|
leptonImg = np.array(struct.unpack("<19200H", leptonData), dtype=np.uint16).reshape((120, 160))
|
|
b = leptonImg.copy()
|
|
for x in range(160):
|
|
for y in range(120):
|
|
b[y][x] = ((leptonImg[y][x]&0xff)<<8) + (leptonImg[y][x]>>8)
|
|
b = cv2.normalize(b, dst=None, alpha=0, beta=255, norm_type=cv2.NORM_MINMAX)
|
|
b = b.astype(np.uint8)
|
|
cv2.imshow("image", cv2.resize(b, (427, 320)))
|
|
leptonVideoWriter.write(b)
|
|
|
|
gridEyeData = f.read(4*64)
|
|
gridEyeImg = np.frombuffer(gridEyeData, dtype=np.uint8).reshape((32, 8))
|
|
reshapeImg = np.zeros((16,16), dtype=np.uint8)
|
|
reshapeImg[0:8, 0:8] = gridEyeImg[0:8]
|
|
reshapeImg[0:8, 8:16] = gridEyeImg[8:16]
|
|
reshapeImg[8:16, 0:8] = gridEyeImg[16:24]
|
|
reshapeImg[8:16, 8:16] = gridEyeImg[24:32]
|
|
reshapeImg = cv2.normalize(reshapeImg, dst=None, alpha=0, beta=255, norm_type=cv2.NORM_MINMAX)
|
|
cv2.imshow("gridEyeImg", cv2.resize(reshapeImg, (320, 320)))
|
|
gridEyeVideoWriter.write(cv2.resize(reshapeImg, (320, 320)))
|
|
key = cv2.waitKey(10)
|
|
if key == ord('q'):
|
|
break
|
|
|
|
leptonVideoWriter.release()
|
|
gridEyeVideoWriter.release()
|
|
cv2.waitKey(0)
|
|
cv2.destroyAllWindows()
|