Ikuinen commited on
Commit
8b7303a
1 Parent(s): a7ac60f

Update grounding_model.py

Browse files
Files changed (1) hide show
  1. grounding_model.py +3 -1
grounding_model.py CHANGED
@@ -11,7 +11,9 @@ from PIL import ImageDraw, ImageFont
11
 
12
  from constants.constant import DARKER_COLOR_MAP, LIGHTER_COLOR_MAP, COLORS
13
  import groundingdino.datasets.transforms as T
14
- from groundingdino import build_groundingdino
 
 
15
  from groundingdino.util.inference import annotate, load_image, predict
16
  from groundingdino.util.utils import clean_state_dict
17
 
 
11
 
12
  from constants.constant import DARKER_COLOR_MAP, LIGHTER_COLOR_MAP, COLORS
13
  import groundingdino.datasets.transforms as T
14
+ # from groundingdino import build_groundingdino
15
+
16
+ from groundingdino.models.GroundingDINO.groundingdino import build_groundingdino
17
  from groundingdino.util.inference import annotate, load_image, predict
18
  from groundingdino.util.utils import clean_state_dict
19