| 13 | from vision_reasoner.models.qwen_vl import QwenVLModel |
| 14 | from vision_reasoner.models.qwen_vl_cot import QwenVLCoTModel |
| 15 | def parse_args(): |
| 16 | parser = argparse.ArgumentParser() |
| 17 | parser.add_argument("--model", type=str, default="vision_reasoner") |
| 18 | parser.add_argument("--model_path", type=str, default="pretrained_models/VisionReasoner-7B") |
| 19 | parser.add_argument("--task_router_model_path", type=str, default="pretrained_models/TaskRouter-1.5B") |
| 20 | parser.add_argument("--segmentation_model_path", type=str, default="facebook/sam2-hiera-large") |
| 21 | parser.add_argument("--output_path", type=str, required=True) |
| 22 | parser.add_argument("--test_data_path", type=str, required=True) |
| 23 | parser.add_argument("--batch_size", type=int, default=1) |
| 24 | |
| 25 | # for parallel evaluation |
| 26 | parser.add_argument("--idx", type=int, required=True) |
| 27 | parser.add_argument("--num_parts", type=int, required=True) |
| 28 | return parser.parse_args() |
| 29 | |
| 30 | def main(): |
| 31 | args = parse_args() |