alex commited on
Commit ·
adc1f0f
1
Parent(s): 38a33d9
log clean up
Browse files- app.py +1 -1
- 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=
|
| 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
|