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 |
|