add log

app.py

@@ -197,27 +197,30 @@ def reconstruct(video_path, conf_thresh, kf_every,
     fps = len(batch) / (end - start)
     print(f'Finished reconstruction for {demo_name}, FPS: {fps:.2f}')
     
-    # Process results
-    pcds = []
-    for j, view in enumerate(batch):
-        image = view['img'].permute(0, 2, 3, 1).cpu().numpy()[0]
-        image = (image + 1) / 2
-        pts = preds[j]['pts3d' if j==0 else 'pts3d_in_other_view'].detach().cpu().numpy()[0]
-        pts_normal = pts2normal(preds[j]['pts3d' if j==0 else 'pts3d_in_other_view'][0]).cpu().numpy()
-        conf = preds[j]['conf'][0].cpu().data.numpy()
-        conf_sig = (conf - 1) / conf
-        if remove_background:
-            mask = generate_mask(image)
-        else:
-            mask = np.ones_like(conf)
+    try:
+        # Process results
+        pcds = []
+        for j, view in enumerate(batch):
+            image = view['img'].permute(0, 2, 3, 1).cpu().numpy()[0]
+            image = (image + 1) / 2
+            pts = preds[j]['pts3d' if j==0 else 'pts3d_in_other_view'].detach().cpu().numpy()[0]
+            pts_normal = pts2normal(preds[j]['pts3d' if j==0 else 'pts3d_in_other_view'][0]).cpu().numpy()
+            conf = preds[j]['conf'][0].cpu().data.numpy()
+            conf_sig = (conf - 1) / conf
+            if remove_background:
+                mask = generate_mask(image)
+            else:
+                mask = np.ones_like(conf)
+                
+            combined_mask = (conf_sig > conf_thresh) & (mask > 0.5)
             
-        combined_mask = (conf_sig > conf_thresh) & (mask > 0.5)
-        
-        pcd = o3d.geometry.PointCloud()
-        pcd.points = o3d.utility.Vector3dVector(pts[combined_mask])
-        pcd.colors = o3d.utility.Vector3dVector(image[combined_mask])
-        pcd.normals = o3d.utility.Vector3dVector(pts_normal[combined_mask])
-        pcds.append(pcd)
+            pcd = o3d.geometry.PointCloud()
+            pcd.points = o3d.utility.Vector3dVector(pts[combined_mask])
+            pcd.colors = o3d.utility.Vector3dVector(image[combined_mask])
+            pcd.normals = o3d.utility.Vector3dVector(pts_normal[combined_mask])
+            pcds.append(pcd)
+    except Exception as e:
+        print(repr(e))
 
     print(f'Finished Process results {demo_name}')