tools: fix replay/camera.py crash (#21881)

* check array size before reshape

* revert

* check length directly
pull/21880/head
Maxime Desroches 4 years ago committed by GitHub
parent 511f5a7589
commit d90188a14e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 5
      tools/replay/camera.py

@ -66,8 +66,9 @@ def ui_thread(addr, frame_address):
flags=cv2.WARP_INVERSE_MAP)
else:
# actually RGB
img = np.frombuffer(yuv_img, dtype=np.uint8).reshape((_FULL_FRAME_SIZE[1], _FULL_FRAME_SIZE[0], 3))
img = img[:, :, ::-1] # Convert BGR to RGB
if len(yuv_img) != 0:
img = np.frombuffer(yuv_img, dtype=np.uint8).reshape((_FULL_FRAME_SIZE[1], _FULL_FRAME_SIZE[0], 3))
img = img[:, :, ::-1] # Convert BGR to RGB
height, width = img.shape[:2]
img_resized = cv2.resize(

Loading…
Cancel
Save