groundingdino

pull/58/head
Kye 2 years ago
parent 4ff7d1661e
commit b6298046ec

@ -29,10 +29,10 @@ from langchain.llms.openai import OpenAI
# Grounding DINO # Grounding DINO
import groundingdino.datasets.transforms as T import groundingdino.datasets.transforms as T
from groundingdino.models import build_model from swarms.agents.models.groundingdino.models import build_model
from groundingdino.util import box_ops from swarms.agents.models.groundingdino.util import box_ops
from groundingdino.util.slconfig import SLConfig from swarms.agents.models.groundingdino.util.slconfig import SLConfig
from groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap from swarms.agents.models.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap
# segment anything # segment anything
from segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator from segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator

Loading…
Cancel
Save