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

Improve opencv

Browse Source
This commit is contained in:
Remi Cadene
2024-07-10 17:43:33 +02:00
parent 2bebdf78a0
commit 7411cf2a7a
2 changed files with 105 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_cameras.py
View File

@@ -1,7 +1,7 @@
import numpy as np
import pytest
from lerobot.common.robot_devices.cameras.opencv import OpenCVCamera
from lerobot.common.robot_devices.cameras.opencv import OpenCVCamera, save_images_from_cameras
from lerobot.common.robot_devices.utils import RobotDeviceAlreadyConnectedError, RobotDeviceNotConnectedError
CAMERA_INDEX = 2
@@ -116,3 +116,7 @@ def test_camera():
with pytest.raises(OSError):
camera.connect()
del camera
def test_save_images_from_cameras(tmpdir):
save_images_from_cameras(tmpdir, record_time_s=1)