Spaces:
Running
on
Zero
Running
on
Zero
fix helper_keyframes
Browse files- visualization/logger.py +14 -6
visualization/logger.py
CHANGED
@@ -143,6 +143,20 @@ class SimulationLogger:
|
|
143 |
if not helper_keyframes:
|
144 |
return
|
145 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
146 |
for keyframe_idx, helper_keyframe in enumerate(helper_keyframes):
|
147 |
try:
|
148 |
position = vector3_to_numpy(helper_keyframe['position'])
|
@@ -168,12 +182,6 @@ class SimulationLogger:
|
|
168 |
timeless=True
|
169 |
)
|
170 |
|
171 |
-
rr.log(
|
172 |
-
f"world/helper_camera_{keyframe_idx}/point",
|
173 |
-
rr.Points3D(colors=[1.0, 1.0, 0.0, 1.0]),
|
174 |
-
timeless=True
|
175 |
-
)
|
176 |
-
|
177 |
except Exception as e:
|
178 |
print(
|
179 |
f"Error logging helper keyframe {keyframe_idx}: {str(e)}")
|
|
|
143 |
if not helper_keyframes:
|
144 |
return
|
145 |
|
146 |
+
|
147 |
+
helper_positions = np.array([
|
148 |
+
vector3_to_numpy(frame['position']) for frame in helper_keyframes
|
149 |
+
])
|
150 |
+
rr.log(
|
151 |
+
"world/helper_keyframes",
|
152 |
+
rr.Points3D(
|
153 |
+
helper_positions,
|
154 |
+
colors=np.full((len(helper_positions), 4),
|
155 |
+
[1.0, 1.0, 0.0, 1.0])
|
156 |
+
),
|
157 |
+
timeless=True
|
158 |
+
)
|
159 |
+
|
160 |
for keyframe_idx, helper_keyframe in enumerate(helper_keyframes):
|
161 |
try:
|
162 |
position = vector3_to_numpy(helper_keyframe['position'])
|
|
|
182 |
timeless=True
|
183 |
)
|
184 |
|
|
|
|
|
|
|
|
|
|
|
|
|
185 |
except Exception as e:
|
186 |
print(
|
187 |
f"Error logging helper keyframe {keyframe_idx}: {str(e)}")
|