Spaces:
Runtime error
Runtime error
update
Browse files- 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.
|
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 |
|