Skip to content

Commit c7a0ba4

Browse files
committed
Cameras folder moved to alignit
1 parent 0e67f68 commit c7a0ba4

2 files changed

Lines changed: 4 additions & 7 deletions

File tree

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,5 @@
1-
#reference to https://github.com/huggingface/lerobot/tree/main
2-
#async_read now returns both rgb and depth images
3-
4-
5-
"""
6-
Provides the RealSenseCamera class for capturing frames from Intel RealSense cameras.
7-
"""
1+
# Reference to https://github.com/huggingface/lerobot/tree/main
2+
# Changed async_read now returns both RGB and depth images
83

94
import logging
105
import time

tests/test_alignnet.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ def test_alignnet_performance():
3838

3939
assert elapsed_time < 0.5
4040

41+
4142
def test_alignnet_with_depth_input():
4243
model = AlignNet(
4344
backbone_name="resnet18",backbone_weights=None,use_vector_input=False,use_depth_input=True, output_dim=7)
@@ -48,6 +49,7 @@ def test_alignnet_with_depth_input():
4849
y = model(x, depth_images=depth)
4950
assert y.shape == (2, 7)
5051

52+
5153
def test_alignnet_with_all_inputs():
5254
# New test for combined inputs
5355
model = AlignNet(backbone_name="efficientnet_b0",backbone_weights=None,use_vector_input=True,use_depth_input=True, output_dim=7)

0 commit comments

Comments
 (0)