Spaces:
Sleeping
Sleeping
sunnychenxiwang
commited on
Commit
•
f9c3654
1
Parent(s):
f16dec2
Update pipeline/tool/object_detetction_model.py
Browse files
pipeline/tool/object_detetction_model.py
CHANGED
@@ -14,9 +14,10 @@ from PIL import Image
|
|
14 |
import numpy as np
|
15 |
from torchvision.ops import box_convert
|
16 |
from pipeline.tool.scene_text_model import *
|
17 |
-
|
18 |
-
|
19 |
-
from
|
|
|
20 |
|
21 |
|
22 |
|
|
|
14 |
import numpy as np
|
15 |
from torchvision.ops import box_convert
|
16 |
from pipeline.tool.scene_text_model import *
|
17 |
+
import sys
|
18 |
+
sys.path.append("pipeline/GroundingDINO")
|
19 |
+
from groundingdino.util.inference import load_model, load_image, predict, annotate
|
20 |
+
# from pipeline.GroundingDINO.groundingdino.util.inference import load_model, load_image, predict, annotate
|
21 |
|
22 |
|
23 |
|