update raft optical flow

2026-01-31 15:22:19 +08:00
parent 2ebbe1a284
commit 58c3d97fab
7 changed files with 117 additions and 141 deletions
@@ -21,7 +21,7 @@ def main():
    # cal_path为校准文件路径, 每一个设备需要创建/使用一个校准文件

    # 方式1: 从视频文件读取
-    sensor = orisys.Sensor("s1.mp4", config_name="./config/prototype_of3.json", cal_path= "./config/dd01a_0.npy", isstitch=True, cuda=False, verbose=True)
+    sensor = orisys.Sensor("s1.mp4", config_name="./config/prototype_of3.json", cal_path= "./config/dd01a_0.npy", isstitch=True, cuda=True, verbose=True)

    # 方式2: 从摄像头读取（取消注释使用） 
    # sensor = orisys.Sensor(0, config_name="./config/prototype_of3.json", cal_path= "./config/cal.npy", isstitch=True, cuda=True, verbose=True)
@@ -58,7 +58,7 @@ def main():
            img,
            flow,
            threshold=4,
-            grid_spacing=10,
+            grid_spacing=40,
            arrow_scale=1.0
        )
        cv2.imshow("Deformation", arrows)