tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add +COLOR_RGB2BGR

Browse Source
This commit is contained in:
Remi Cadene
2024-09-27 12:11:48 +02:00
parent a7350d9b65
commit bf7e906b70
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/mock_opencv.py
View File

@@ -5,6 +5,7 @@ import numpy as np
CAP_PROP_FPS = 5 CAP_PROP_FPS = 5
CAP_PROP_FRAME_WIDTH = 3 CAP_PROP_FRAME_WIDTH = 3
CAP_PROP_FRAME_HEIGHT = 4 CAP_PROP_FRAME_HEIGHT = 4
COLOR_RGB2BGR = 4
COLOR_BGR2RGB = 4 COLOR_BGR2RGB = 4
@@ -14,7 +15,7 @@ def _generate_image(width: int, height: int):
def cvtColor(color_image, color_convertion): # noqa: N802 def cvtColor(color_image, color_convertion): # noqa: N802
if color_convertion == COLOR_BGR2RGB: if color_convertion in [COLOR_RGB2BGR, COLOR_BGR2RGB]:
return color_image[:, :, [2, 1, 0]] return color_image[:, :, [2, 1, 0]]
else: else:
raise NotImplementedError(color_convertion) raise NotImplementedError(color_convertion)