From 6fee80f731c72727955a356d19ba695c6d68a67f Mon Sep 17 00:00:00 2001 From: neosouwchuan <55700126+neosouwchuan@users.noreply.github.com> Date: Thu, 13 Jun 2024 06:32:25 +0000 Subject: [PATCH] Add VLM models for Finals --- vlm/src/VLMManager.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vlm/src/VLMManager.py b/vlm/src/VLMManager.py index 8ca3af4..10c1f28 100644 --- a/vlm/src/VLMManager.py +++ b/vlm/src/VLMManager.py @@ -24,9 +24,9 @@ def __init__(self): ]) print(os.getcwd()) print([f for f in os.listdir('.') if os.path.isfile(f)]) - self.clipmodel= torch.load(path.join(path.dirname(path.abspath(__file__)), "clip_ft_2.pt")) + self.clipmodel= torch.load(path.join(path.dirname(path.abspath(__file__)), "clip_ft_final.pt")) self.objects = ["cargo aircraft","light aircraft","commercial aircraft","drone","missile","helicopter","fighter jet","fighter plane"] - self.model = YOLOWorld(path.join(path.dirname(path.abspath(__file__)), "augmentedbestv2.pt")).to(self.device) + self.model = YOLOWorld(path.join(path.dirname(path.abspath(__file__)), "highdegree60.pt")).to(self.device) for i in self.clipmodel.parameters(): i.requires_grad=False for i in self.model.parameters():