Update eval_wrapper/eval.py

eval_wrapper/eval.py

@@ -17,9 +17,9 @@ from utils import misc
 import torch.distributed as dist
 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 filter_all_masks
+# from utils.viz import just_load_viz
+from utils.geometry import compute_pointmap_torch
+from eval_wrapper.eval_utils import filter_all_masks
 from huggingface_hub import hf_hub_download
 
 class EvalWrapper(torch.nn.Module):