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

Made script to save camera images instead of these funcitons existing in the camera driver classes.

Browse Source 
Fixed test_cameras.py script so it passes.
This commit is contained in:
jess-moss
2024-09-20 14:53:15 -05:00
parent be5336f9d3
commit 34dcd05a96
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_cameras.py
View File

@@ -7,12 +7,15 @@ pytest -sx tests/test_cameras.py::test_camera
``` ```
""" """
from pathlib import Path
import numpy as np import numpy as np
import pytest import pytest
from lerobot import available_robots from lerobot import available_robots
from lerobot.common.robot_devices.cameras.opencv import OpenCVCamera, save_images_from_cameras from lerobot.common.robot_devices.cameras.opencv import OpenCVCamera
from lerobot.common.robot_devices.utils import RobotDeviceAlreadyConnectedError, RobotDeviceNotConnectedError from lerobot.common.robot_devices.utils import RobotDeviceAlreadyConnectedError, RobotDeviceNotConnectedError
from lerobot.scripts.save_images_from_cameras import save_images_from_cameras
from tests.utils import require_robot from tests.utils import require_robot
CAMERA_INDEX = 2 CAMERA_INDEX = 2
@@ -131,7 +134,8 @@ def test_camera(request, robot_type):
del camera del camera
@pytest.mark.parametrize("robot_type", available_robots) def test_save_images_from_cameras():
@require_robot # Set the path relative to the project root (lerobot) where 'outputs' already exists
def test_save_images_from_cameras(tmpdir, request, robot_type): images_dir = Path("outputs/images_from_opencv_cameras")
save_images_from_cameras(tmpdir, record_time_s=1) images_dir.mkdir(parents=True, exist_ok=True)
save_images_from_cameras(driver="opencv", images_dir=images_dir)