bp

eval_wrapper/eval.py

@@ -19,7 +19,7 @@ from utils.collate import collate
 from engine import eval_model
 from utils.viz import just_load_viz
 from utils.geometry import compute_pointmap_torch
-from eval_wrapper.eval_utils import npy2ply, filter_all_masks
+from eval_wrapper.eval_utils import filter_all_masks
 from huggingface_hub import hf_hub_download
 
 class EvalWrapper(torch.nn.Module):

eval_wrapper/eval_utils.py

@@ -2,7 +2,6 @@ import numpy as np
 import matplotlib.pyplot as plt
 from scipy.stats import norm, lognorm
 import torch
-import open3d as o3d
 
 def colorize_points_with_turbo_all_dims(points, method='norm',cmap='turbo'):
     """
@@ -38,21 +37,6 @@ def colorize_points_with_turbo_all_dims(points, method='norm',cmap='turbo'):
 
     return colors
 
-def npy2ply(points,colors=None,normals=None):
-  cloud = o3d.geometry.PointCloud()
-  cloud.points = o3d.utility.Vector3dVector(points.astype(np.float64))
-
-  # compute the normals
-  if colors is not None:
-    if colors.max()>1:
-      colors = colors/255.0
-    cloud.colors = o3d.utility.Vector3dVector(colors.astype(np.float64))
-  else:
-    colors = colorize_points_with_turbo_all_dims(points)
-    cloud.colors = o3d.utility.Vector3dVector(colors.astype(np.float64))
-  if normals is not None:
-    cloud.normals = o3d.utility.Vector3dVector(normals.astype(np.float64))
-  return cloud
 
 def transform_pointmap(pointmap_cam,c2w):
     # pointmap: shape H x W x 3