Update eval_wrapper/eval.py

eval_wrapper/eval.py

@@ -10,17 +10,17 @@ sys.path.append(current_dir)
 
 from eval_wrapper.sample_poses import pointmap_to_poses
 from utils.fusion import fuse_batch
-# from models.rayquery import *
-# from models.losses import *
-# import argparse
-# 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 models.rayquery import *
+from models.losses import *
+import argparse
+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
 
 class EvalWrapper(torch.nn.Module):
     def __init__(self,checkpoint_path,distributed=False,device="cpu",dtype=torch.float32,**kwargs):