diff --git a/src/data/basic_dataset.py b/src/data/basic_dataset.py index a7a6cc4..0d8fef4 100644 --- a/src/data/basic_dataset.py +++ b/src/data/basic_dataset.py @@ -124,7 +124,9 @@ def __init__( else None ) - camera_calibration = scene_obj[key_frame_id]["camera_calibration"] + camera_calibration = None + if split=="val" and add_kois and add_bev: + camera_calibration = scene_obj[key_frame_id]["camera_calibration"] qas = scene_obj[key_frame_id]["QA"] diff --git a/src/models/qwen_vl_inference.py b/src/models/qwen_vl_inference.py index 6487604..342c497 100644 --- a/src/models/qwen_vl_inference.py +++ b/src/models/qwen_vl_inference.py @@ -17,7 +17,7 @@ class QwenVLInferenceEngine(BaseInferenceEngine): def __init__( self, - processor_path: str = "Qwen/Qwen2.5-VL-3B-Instruct", + processor_path: str = "Qwen/Qwen2.5-VL-7B-Instruct", model_path: Optional[str] = None, use_4bit: bool = False, torch_dtype: Optional[torch.dtype] = None,