wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add compatibility with camera index as port

This commit is contained in:
Remi Cadene 2024-07-19 11:31:32 +02:00
parent 1664dac300
commit b0f4f4f9a1
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lerobot/common/robot_devices/cameras/opencv.py
View File

@ -182,7 +182,7 @@ class OpenCVCamera:
``` ```
""" """
def __init__(self, camera_index: int, config: OpenCVCameraConfig | None = None, **kwargs): def __init__(self, camera_index: int | str, config: OpenCVCameraConfig | None = None, **kwargs):
if config is None: if config is None:
config = OpenCVCameraConfig() config = OpenCVCameraConfig()
# Overwrite config arguments using kwargs # Overwrite config arguments using kwargs
@ -194,11 +194,6 @@ class OpenCVCamera:
self.height = config.height self.height = config.height
self.color_mode = config.color_mode self.color_mode = config.color_mode
if not isinstance(self.camera_index, int):
raise ValueError(
f"Camera index must be provided as an int, but {self.camera_index} was given instead."
)
self.camera = None self.camera = None
self.is_connected = False self.is_connected = False
self.thread = None self.thread = None