alex commited on
Commit
adc1f0f
·
1 Parent(s): 38a33d9

log clean up

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. src/pixel3dmm/tracking/tracker.py +1 -3
app.py CHANGED
@@ -261,7 +261,7 @@ def step4_track(session_id):
261
  return "✅ Pipeline complete!", image, gr.update(interactive=True)
262
 
263
  # New: run all steps sequentially
264
- @spaces.GPU(duration=120)
265
  def generate_results_and_mesh(image, session_id=None):
266
  """
267
  Process an input image through a 3D reconstruction pipeline and return the intermediate outputs and mesh file.
 
261
  return "✅ Pipeline complete!", image, gr.update(interactive=True)
262
 
263
  # New: run all steps sequentially
264
+ @spaces.GPU(duration=150)
265
  def generate_results_and_mesh(image, session_id=None):
266
  """
267
  Process an input image through a 3D reconstruction pipeline and return the intermediate outputs and mesh file.
src/pixel3dmm/tracking/tracker.py CHANGED
@@ -675,7 +675,7 @@ class Tracker(object):
675
  self.r_mvps[serial] = r_mvps
676
 
677
  loss = all_loss.item()
678
- t.set_description(f'Loss for camera {loss:.4f}')
679
  self.frame += 1
680
  #if k % 100 == 0:
681
  # self.checkpoint(batch, visualizations=[[View.GROUND_TRUTH, View.LANDMARKS, View.SHAPE_OVERLAY, View.COLOR_OVERLAY]], frame_dst='/camera', save=False, dump_directly=True, is_camera=True)
@@ -1156,8 +1156,6 @@ class Tracker(object):
1156
  # frame_dst='/debug_joint', save=False, dump_directly=True, timestep=p, selected_frames=selected_frames, is_final=True)
1157
  self.frame += 1
1158
 
1159
- iterator.set_description(f'Timestep {save_timestep}; Loss {all_loss.item():.4f}')
1160
-
1161
  #if n_steps_stagnant > 35 and not is_joint:
1162
  # print('Early Stopping, go to next frame!')
1163
  # #break
 
675
  self.r_mvps[serial] = r_mvps
676
 
677
  loss = all_loss.item()
678
+ # t.set_description(f'Loss for camera {loss:.4f}')
679
  self.frame += 1
680
  #if k % 100 == 0:
681
  # self.checkpoint(batch, visualizations=[[View.GROUND_TRUTH, View.LANDMARKS, View.SHAPE_OVERLAY, View.COLOR_OVERLAY]], frame_dst='/camera', save=False, dump_directly=True, is_camera=True)
 
1156
  # frame_dst='/debug_joint', save=False, dump_directly=True, timestep=p, selected_frames=selected_frames, is_final=True)
1157
  self.frame += 1
1158
 
 
 
1159
  #if n_steps_stagnant > 35 and not is_joint:
1160
  # print('Early Stopping, go to next frame!')
1161
  # #break