Add VLM models for Finals

qitianshi · Jun 13, 2024 · 6fee80f · 6fee80f
1 parent 2e6dc91
commit 6fee80f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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():