update app.py

Grounding_DINO/groundingdino/datasets/transforms.py

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

Grounding_DINO/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.util import box_ops, get_tokenlizer
-from groundingdino.util.misc import (
+from Grounding_DINO.groundingdino.util import box_ops, get_tokenlizer
+from Grounding_DINO.groundingdino.util.misc import (
     NestedTensor,
     accuracy,
     get_world_size,
@@ -33,9 +33,9 @@ from groundingdino.util.misc import (
     is_dist_avail_and_initialized,
     nested_tensor_from_tensor_list,
 )
-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 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 ..registry import MODULE_BUILD_FUNCS
 from .backbone import build_backbone

app.py

@@ -1,20 +1,21 @@
 
 import subprocess, os, sys
 
-if 0==1:
-    result = subprocess.run(['pip', 'install', '-e', 'segment_anything'], check=True)
-    print(f'liuyz_install segment_anything result = {result}')
-    result = subprocess.run(['pip', 'install', '-e', 'Grounding_DINO'], check=True)
-    print(f'liuyz_install Grounding_DINO result = {result}')
+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}')
 
 result = subprocess.run(['pip', 'list'], check=True)
 print(f'liuyz_pip list result = {result}')
 
-if not os.path.exists('/sam_vit_h_4b8939.pth'):
+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}')    
 
-
 import gradio as gr
 
 import argparse
@@ -26,10 +27,10 @@ from PIL import Image, ImageDraw, ImageFont
 
 # Grounding DINO
 import Grounding_DINO.groundingdino.datasets.transforms as T
-from groundingdino.models import build_model
-from groundingdino.util import box_ops
-from groundingdino.util.slconfig import SLConfig
-from groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap
+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
 
 # segment anything
 from segment_anything import build_sam, SamPredictor 

requirements.txt

@@ -20,6 +20,6 @@ transformers
 yapf
 numba
 segment_anything
-git+https://github.com/IDEA-Research/GroundingDINO
+# git+https://github.com/IDEA-Research/GroundingDINO
 # GroundingDINO
 # segment_anything