abreza commited on
Commit
15db7ce
1 Parent(s): 50e34ab

enhance logging for helper keyframes by adding timeless parameter and reorganizing camera trajectory logging

Browse files
visualization/logger.py CHANGED
@@ -154,7 +154,8 @@ class SimulationLogger:
154
  translation=position,
155
  rotation=rr.Quaternion(xyzw=rotation_q),
156
  scale=(1.0, 1.0, 1.0)
157
- )
 
158
  )
159
 
160
  rr.log(
@@ -163,8 +164,17 @@ class SimulationLogger:
163
  image_from_camera=self.K,
164
  width=1920,
165
  height=1080,
166
- color=(1.0, 1.0, 0.0, 1.0)
167
- )
 
 
 
 
 
 
 
 
 
168
  )
169
 
170
  except Exception as e:
 
154
  translation=position,
155
  rotation=rr.Quaternion(xyzw=rotation_q),
156
  scale=(1.0, 1.0, 1.0)
157
+ ),
158
+ timeless=True
159
  )
160
 
161
  rr.log(
 
164
  image_from_camera=self.K,
165
  width=1920,
166
  height=1080,
167
+ ),
168
+ timeless=True
169
+ )
170
+
171
+ rr.log(
172
+ "world/helper_keyframes/camera_point_{keyframe_idx}",
173
+ rr.Points3D(
174
+ position,
175
+ colors=np.full((len(position), 4), [1.0, 1.0, 0.0, 1.0])
176
+ ),
177
+ timeless=True
178
  )
179
 
180
  except Exception as e:
visualization/visualizer.py CHANGED
@@ -26,11 +26,12 @@ def visualize_simulation(file, simulation_index: Optional[int]) -> Optional[str]
26
  logger = SimulationLogger()
27
  logger.log_metadata(simulation['instructions'])
28
  logger.log_subjects(simulation['subjects'])
29
- logger.log_camera_trajectory(simulation['cameraFrames'])
30
- logger.log_camera_frames(simulation['cameraFrames'])
31
 
32
  if 'helper_keyframes' in simulation:
33
  logger.log_helper_keyframes(simulation['helper_keyframes'])
 
 
 
34
 
35
  rr.save(rrd_path)
36
 
 
26
  logger = SimulationLogger()
27
  logger.log_metadata(simulation['instructions'])
28
  logger.log_subjects(simulation['subjects'])
 
 
29
 
30
  if 'helper_keyframes' in simulation:
31
  logger.log_helper_keyframes(simulation['helper_keyframes'])
32
+
33
+ logger.log_camera_trajectory(simulation['cameraFrames'])
34
+ logger.log_camera_frames(simulation['cameraFrames'])
35
 
36
  rr.save(rrd_path)
37