| 17 | from vision_reasoner.models.visurf_model import ViSurfModel |
| 18 | |
| 19 | def parse_args(): |
| 20 | parser = argparse.ArgumentParser() |
| 21 | parser.add_argument("--model", type=str, default="vision_reasoner") |
| 22 | parser.add_argument("--model_path", type=str, default="pretrained_models/VisionReasoner-7B") |
| 23 | parser.add_argument("--task_router_model_path", type=str, default="pretrained_models/TaskRouter-1.5B") |
| 24 | parser.add_argument("--segmentation_model_path", type=str, default="facebook/sam2-hiera-large") |
| 25 | parser.add_argument("--output_path", type=str, required=True) |
| 26 | parser.add_argument("--test_data_path", type=str, required=True) |
| 27 | parser.add_argument("--batch_size", type=int, default=1) |
| 28 | |
| 29 | # for parallel evaluation |
| 30 | parser.add_argument("--idx", type=int, required=True) |
| 31 | parser.add_argument("--num_parts", type=int, required=True) |
| 32 | return parser.parse_args() |
| 33 | |
| 34 | def compute_iou(mask1, mask2): |
| 35 | intersection = np.logical_and(mask1, mask2).sum() |