Update eval_wrapper/eval.py

eval_wrapper/eval.py

@@ -17,10 +17,10 @@ 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 huggingface_hub import hf_hub_download
+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):
     def __init__(self,checkpoint_path,distributed=False,device="cpu",dtype=torch.float32,**kwargs):