diff --git a/swarms/agents/multi_modal_visual_agent.py b/swarms/agents/multi_modal_visual_agent.py index 0fba3333..b97c6c65 100644 --- a/swarms/agents/multi_modal_visual_agent.py +++ b/swarms/agents/multi_modal_visual_agent.py @@ -28,11 +28,11 @@ from langchain.chains.conversation.memory import ConversationBufferMemory from langchain.llms.openai import OpenAI # Grounding DINO -import swarms.agents.models.groundingdino.groundingdino.datasets.transforms as T -from swarms.agents.models.groundingdino.groundingdino.models import build_model -from swarms.agents.models.groundingdino.groundingdino.util import box_ops -from swarms.agents.models.groundingdino.groundingdino.util.slconfig import SLConfig -from swarms.agents.models.groundingdino.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap +import swarms.agents.models.groundingdino.datasets.transforms as T +from swarms.agents.models.groundingdino.models import build_model +from swarms.agents.models.groundingdino.util import box_ops +from swarms.agents.models.groundingdino.util.slconfig import SLConfig +from swarms.agents.models.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap # segment anything from segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator @@ -1025,7 +1025,7 @@ class Text2Box: print(f"Initializing ObjectDetection to {device}") self.device = device self.torch_dtype = torch.float16 if 'cuda' in device else torch.float32 - self.model_checkpoint_path = os.path.join("checkpoints","groundingdino"groundingdino.) + self.model_checkpoint_path = os.path.join("checkpoints","groundingdino") self.model_config_path = os.path.join("checkpoints","grounding_config.py") self.download_parameters() self.box_threshold = 0.3 @@ -1033,10 +1033,10 @@ class Text2Box: self.grounding = (self.load_model()).to(self.device) def download_parameters(self): - url = "https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_groundingdino.swint_ogc.pth" + url = "https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth" if not os.path.exists(self.model_checkpoint_path): wget.download(url,out=self.model_checkpoint_path) - config_url = "https://raw.githubusercontent.com/IDEA-Research/GroundingDINO/main/groundingdino/groundingdino.config/GroundingDINO_SwinT_OGC.py" + config_url = "https://raw.githubusercontent.com/IDEA-Research/GroundingDINO/main/groundingdino/config/GroundingDINO_SwinT_OGC.py" if not os.path.exists(self.model_config_path): wget.download(config_url,out=self.model_config_path) def load_image(self,image_path):