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 Grounding_DINO.groundingdino.util.box_ops import box_xyxy_to_cxcywh
-from Grounding_DINO.groundingdino.util.misc import interpolate
+from GroundingDINO.groundingdino.util.box_ops import box_xyxy_to_cxcywh
+from GroundingDINO.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 Grounding_DINO.groundingdino.util import box_ops, get_tokenlizer
-from Grounding_DINO.groundingdino.util.misc import (
+from GroundingDINO.groundingdino.util import box_ops, get_tokenlizer
+from GroundingDINO.groundingdino.util.misc import (
     NestedTensor,
     accuracy,
     get_world_size,
@@ -33,9 +33,9 @@ from Grounding_DINO.groundingdino.util.misc import (
     is_dist_avail_and_initialized,
     nested_tensor_from_tensor_list,
 )
-from Grounding_DINO.groundingdino.util.utils import get_phrases_from_posmap
-from Grounding_DINO.groundingdino.util.visualizer import COCOVisualizer
-from Grounding_DINO.groundingdino.util.vl_utils import create_positive_map_from_span
+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 ..registry import MODULE_BUILD_FUNCS
 from .backbone import build_backbone

app.py

@@ -1,20 +1,14 @@
 
 import subprocess, os, sys
 
-if 0==0:
-    # result = subprocess.run(['pip', 'install', '-e', 'segment_anything'], check=True)
-    # print(f'liuyz_install segment_anything result = {result}')
-
-    sys.path.insert(0, './Grounding_DINO')
-    # result = subprocess.run(['pip', 'install', '-e', 'Grounding_DINO'], check=True)
-    # print(f'liuyz_install Grounding_DINO result = {result}')
+sys.path.insert(0, './GroundingDINO')
 
 result = subprocess.run(['pip', 'list'], check=True)
-print(f'liuyz_pip list result = {result}')
+print(f'pip list result = {result}')
 
 if not os.path.exists('./sam_vit_h_4b8939.pth'):
     result = subprocess.run(['wget', 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth'], check=True)
-    print(f'liuyz_wget sam_vit_h_4b8939.pth result = {result}')    
+    print(f'wget sam_vit_h_4b8939.pth result = {result}')    
 
 import gradio as gr
 
@@ -26,11 +20,11 @@ import torch
 from PIL import Image, ImageDraw, ImageFont
 
 # Grounding DINO
-import Grounding_DINO.groundingdino.datasets.transforms as T
-from Grounding_DINO.groundingdino.models import build_model
-from Grounding_DINO.groundingdino.util import box_ops
-from Grounding_DINO.groundingdino.util.slconfig import SLConfig
-from Grounding_DINO.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap
+import GroundingDINO.groundingdino.datasets.transforms as T
+from GroundingDINO.groundingdino.models import build_model
+from GroundingDINO.groundingdino.util import box_ops
+from GroundingDINO.groundingdino.util.slconfig import SLConfig
+from GroundingDINO.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap
 
 # segment anything
 from segment_anything import build_sam, SamPredictor