mawei/onepilot
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* check array size before reshape

* revert

* check length directly
old-commit-hash: d90188a14e7f9a8e07850b4577cdaca037aa4903
This commit is contained in:
Maxime Desroches
2021-08-09 03:43:11 -04:00
committed by GitHub
parent e43bc042a7
commit 202af3bb4e
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tools/replay/camera.py
View File

@@ -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(