Improve VLM model with augmented training dataset

vlm/src/VLMManager.py

@@ -26,7 +26,7 @@ def __init__(self):
         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.objects = ["cargo aircraft","light aircraft","commercial aircraft","drone","missile","helicopter","fighter jet","fighter plane"]
-        self.model = YOLOWorld(path.join(path.dirname(path.abspath(__file__)), "800allbest.pt")).to(self.device)
+        self.model = YOLOWorld(path.join(path.dirname(path.abspath(__file__)), "augmentedbestv2.pt")).to(self.device)
         for i in self.clipmodel.parameters():
             i.requires_grad=False
         for i in self.model.parameters():