clean up

swarms/agents/workers/models/__init__.py

@@ -3,3 +3,27 @@
 # from .GroundingDINO.groundingdino.util import box_ops, SLConfig
 # from .GroundingDINO.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap
 # from .segment_anything.segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator
+from swarms.agents.workers.models.GroundingDINO.groundingdino.datasets.transforms import (
+    Compose,
+    Normalize,
+    ToTensor,
+    crop,
+    hflip,
+    resize,
+    pad,
+    ResizeDebug,
+    RandomCrop,
+    RandomSizeCrop,
+    CenterCrop,
+    RandomHorizontalFlip,
+    RandomResize,
+    RandomPad,
+    RandomSelect
+)
+
+from swarms.agents.workers.models.GroundingDINO.groundingdino.models import build_model
+from swarms.agents.workers.models.GroundingDINO.groundingdino.util import box_ops
+from swarms.agents.workers.models.GroundingDINO.groundingdino.util.slconfig import SLConfig
+from swarms.agents.workers.models.GroundingDINO.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap
+
+from swarms.agents.workers.models.segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator

swarms/agents/workers/multi_modal/multi_modal_agent.py

@@ -33,7 +33,7 @@ from langchain.llms.openai import OpenAI
 
 # Grounding DINO
 # import groundingdino.datasets.transforms as T
-from swarms.agents.workers.models.GroundingDINO.groundingdino.datasets.transforms import (
+from swarms.agents.workers.models import (
     Compose,
     Normalize,
     ToTensor,
@@ -50,10 +50,11 @@ from swarms.agents.workers.models.GroundingDINO.groundingdino.datasets.transform
     RandomPad,
     RandomSelect
 )
-from swarms.agents.workers.models.GroundingDINO.groundingdino.models import build_model
-from swarms.agents.workers.models.GroundingDINO.groundingdino.util import box_ops
-from swarms.agents.workers.models.GroundingDINO.groundingdino.util.slconfig import SLConfig
-from swarms.agents.workers.models.GroundingDINO.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap
+
+from swarms.agents.workers.models import build_model
+from swarms.agents.workers.models  import box_ops
+from swarms.agents.workers.models import SLConfig
+from swarms.agents.workers.models import clean_state_dict, get_phrases_from_posmap
 
 from swarms.agents.workers.models.segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator
 import cv2