make camera work and fix test

uci-uav-forge · Jun 25, 2024 · 163af08 · 163af08
1 parent 397b912
commit 163af08
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/scripts/perception_clients/test_only_camera.py b/scripts/perception_clients/test_only_camera.py
@@ -1,9 +1,11 @@
 from uavf_2024.imaging import Camera
+import time
 import cv2 as cv
 import os
 from tqdm import tqdm
 
 cam = Camera()
+cam.start_recording()
 
 cam.setAbsoluteZoom(1)
 cam.request_center()
@@ -17,5 +19,8 @@
     img = cam.get_latest_image()
     if img is not None:
         img.save(f"snapshots/img_{i}.jpg")
+    else:
+        print("image is none")
+    time.sleep(0.1)
 
 cam.disconnect()
diff --git a/uavf_2024/imaging/camera_control.py b/uavf_2024/imaging/camera_control.py
@@ -2,7 +2,8 @@
 from pathlib import Path
 import threading
 import time
-from siyi_sdk import MockSIYISTREAM as SIYISTREAM, MockSIYISDK as SIYISDK
+# from siyi_sdk import MockSIYISTREAM as SIYISTREAM, MockSIYISDK as SIYISDK
+from siyi_sdk import SIYISTREAM, SIYISDK
 from uavf_2024.imaging.imaging_types import Image, HWC
 from scipy.spatial.transform import Rotation
 import numpy as np