liuyizhang commited on
Commit
00f982a
1 Parent(s): c7f215e

update groundingdino.py

Browse files
GroundingDINO/groundingdino/models/GroundingDINO/groundingdino.py CHANGED
@@ -23,8 +23,8 @@ from torch import nn
23
  from torchvision.ops.boxes import nms
24
  from transformers import AutoTokenizer, BertModel, BertTokenizer, RobertaModel, RobertaTokenizerFast
25
 
26
- from groundingdino.util import box_ops, get_tokenlizer
27
- from groundingdino.util.misc import (
28
  NestedTensor,
29
  accuracy,
30
  get_world_size,
@@ -33,9 +33,9 @@ from groundingdino.util.misc import (
33
  is_dist_avail_and_initialized,
34
  nested_tensor_from_tensor_list,
35
  )
36
- from groundingdino.util.utils import get_phrases_from_posmap
37
- from groundingdino.util.visualizer import COCOVisualizer
38
- from groundingdino.util.vl_utils import create_positive_map_from_span
39
 
40
  from ..registry import MODULE_BUILD_FUNCS
41
  from .backbone import build_backbone
 
23
  from torchvision.ops.boxes import nms
24
  from transformers import AutoTokenizer, BertModel, BertTokenizer, RobertaModel, RobertaTokenizerFast
25
 
26
+ from GroundingDINO.groundingdino.util import box_ops, get_tokenlizer
27
+ from GroundingDINO.groundingdino.util.misc import (
28
  NestedTensor,
29
  accuracy,
30
  get_world_size,
 
33
  is_dist_avail_and_initialized,
34
  nested_tensor_from_tensor_list,
35
  )
36
+ from GroundingDINO.groundingdino.util.utils import get_phrases_from_posmap
37
+ from GroundingDINO.groundingdino.util.visualizer import COCOVisualizer
38
+ from GroundingDINO.groundingdino.util.vl_utils import create_positive_map_from_span
39
 
40
  from ..registry import MODULE_BUILD_FUNCS
41
  from .backbone import build_backbone