update app.py

GroundingDINO/groundingdino/datasets/transforms.py

@@ -10,8 +10,8 @@ import torch
 import torchvision.transforms as T
 import torchvision.transforms.functional as F
 
-from GroundingDINO.groundingdino.util.box_ops import box_xyxy_to_cxcywh
-from GroundingDINO.groundingdino.util.misc import interpolate
+from groundingdino.util.box_ops import box_xyxy_to_cxcywh
+from groundingdino.util.misc import interpolate
 
 
 def crop(image, target, region):

GroundingDINO/groundingdino/models/GroundingDINO/groundingdino.py

@@ -23,8 +23,8 @@ from torch import nn
 from torchvision.ops.boxes import nms
 from transformers import AutoTokenizer, BertModel, BertTokenizer, RobertaModel, RobertaTokenizerFast
 
-from GroundingDINO.groundingdino.util import box_ops, get_tokenlizer
-from GroundingDINO.groundingdino.util.misc import (
+from groundingdino.util import box_ops, get_tokenlizer
+from groundingdino.util.misc import (
     NestedTensor,
     accuracy,
     get_world_size,
@@ -33,9 +33,9 @@ from GroundingDINO.groundingdino.util.misc import (
     is_dist_avail_and_initialized,
     nested_tensor_from_tensor_list,
 )
-from GroundingDINO.groundingdino.util.utils import get_phrases_from_posmap
-from GroundingDINO.groundingdino.util.visualizer import COCOVisualizer
-from GroundingDINO.groundingdino.util.vl_utils import create_positive_map_from_span
+from groundingdino.util.utils import get_phrases_from_posmap
+from groundingdino.util.visualizer import COCOVisualizer
+from groundingdino.util.vl_utils import create_positive_map_from_span
 
 from ..registry import MODULE_BUILD_FUNCS
 from .backbone import build_backbone

app.py

@@ -1,11 +1,19 @@
-import os
-os.system("pip install -e segment_anything")
-os.system("pip install -e GroundingDINO")
+
+import subprocess
+
+result = subprocess.run(['pip', 'install', '-e', 'segment_anything'], check=True)
+print(f'install segment_anything result = {result}')
+result = subprocess.run(['pip', 'install', '-e', 'GroundingDINO'], check=True)
+print(f'install GroundingDINO result = {result}')
+
+# os.system("pip install -e segment_anything")
+# os.system("pip install -e GroundingDINO")
 
 import gradio as gr
 
 import argparse
 import copy
+import os
 
 import numpy as np
 import torch