fix numpy error

shunsukesaito · Apr 9, 2024 · 6acce5d · 6acce5d
1 parent f0a9c99
commit 6acce5d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/apps/eval.py b/apps/eval.py
@@ -99,6 +99,7 @@ def eval(self, data, use_octree=False):
             if self.netC:
                 self.netC.eval()
             save_path = '%s/%s/result_%s.obj' % (opt.results_path, opt.name, data['name'])
+            print('save path:',save_path)
             if self.netC:
                 gen_mesh_color(opt, self.netG, self.netC, self.cuda, data, save_path, use_octree=use_octree)
             else:
@@ -118,6 +119,7 @@ def eval(self, data, use_octree=False):
         try:
             print(image_path, mask_path)
             data = evaluator.load_image(image_path, mask_path)
+            print('start eval')
             evaluator.eval(data, True)
         except Exception as e:
            print("error:", e.args)
diff --git a/lib/sdf.py b/lib/sdf.py
@@ -55,8 +55,8 @@ def eval_grid_octree(coords, eval_func,
 
     sdf = np.zeros(resolution)
 
-    dirty = np.ones(resolution, dtype=np.bool)
-    grid_mask = np.zeros(resolution, dtype=np.bool)
+    dirty = np.ones(resolution, dtype=bool)
+    grid_mask = np.zeros(resolution, dtype=bool)
 
     reso = resolution[0] // init_resolution