update: fixed a bug

changhaonan · Sep 11, 2023 · 589f960 · 589f960
1 parent f8842c4
commit 589f960
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lgmcts/scripts/data_generation/gen_strdiff.py b/lgmcts/scripts/data_generation/gen_strdiff.py
@@ -110,7 +110,8 @@ def _generate_data_for_one_task(
             # normalize depth to fit in structFormer
             depth_tensor = depth_tensor * 20.0
             # depth_min & depth_max
-            depth_min = np.min(depth_tensor) * np.ones([2,], dtype=np.float32)
+            depth_min = np.min(depth_tensor) * np.ones([1,], dtype=np.float32)
+            depth_max = np.max(depth_tensor) * np.ones([1,], dtype=np.float32)
             f.create_dataset("depth_min", data=depth_min)
             f.create_dataset("depth_max", data=depth_max)
             # normalize depth