Spaces:

TencentARC
/

GeometryCrafter

Running on Zero

slothfulxtx commited on 20 days ago

Commit

34120aa

1 Parent(s): 34a1c67

fix npz name bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -172,7 +172,7 @@ def infer_geometry(
         np.savez_compressed(
             output_npz_path,
             point_map=point_maps.cpu().numpy().astype(np.float16),
-            valid_mask=valid_masks.cpu().numpy().astype(np.bool_)
         )
         output_disp_path = Path(save_folder, run_id, f'disp.mp4')

         np.savez_compressed(
             output_npz_path,
             point_map=point_maps.cpu().numpy().astype(np.float16),
+            mask=valid_masks.cpu().numpy().astype(np.bool_)
         )
         output_disp_path = Path(save_folder, run_id, f'disp.mp4')