ikuinen99 commited on
Commit
6abcd14
1 Parent(s): a5413d0
Files changed (1) hide show
  1. grounding_model.py +2 -2
grounding_model.py CHANGED
@@ -9,10 +9,10 @@ from segment_anything import build_sam, SamPredictor
9
  from segment_anything.utils.amg import remove_small_regions
10
  from PIL import ImageDraw, ImageFont
11
 
12
- import groundingdino.util.transforms as T
13
  from constants.constant import DARKER_COLOR_MAP, LIGHTER_COLOR_MAP, COLORS
 
14
  from groundingdino import build_groundingdino
15
- from groundingdino.util.predict import predict
16
  from groundingdino.util.utils import clean_state_dict
17
 
18
 
 
9
  from segment_anything.utils.amg import remove_small_regions
10
  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
 
18