update

Dockerfile

@@ -1,2 +0,0 @@
-ENV CUDA_HOME /usr/local/cuda-11.3/
-RUN pip install git+https://github.com/IDEA-Research/GroundingDINO.git

app.py

@@ -67,7 +67,7 @@ args = parse_args()
 
 assert args.dummy or (args.cfg_path is not None), "Invalid Config! Set --dummy or configurate the cfg_path!"
 
-device = 'cuda:{}'.format(args.gpu_id) if torch.cuda.is_available() else 'cpu'
+device = 'cuda'.format(args.gpu_id) if torch.cuda.is_available() else 'cpu'
 
 if not args.dummy:
     cfg = Config(args)

groundingdino/models/GroundingDINO/ms_deform_attn.py

@@ -26,11 +26,11 @@ from torch.autograd.function import once_differentiable
 from torch.nn.init import constant_, xavier_uniform_
 
 try:
-    from groundingdino import _C
+    # from groundingdino import _C
     # from mmcv.utils import ext_loader
 
-    # _C = ext_loader.load_ext(
-    #     '_ext', ['ms_deform_attn_backward', 'ms_deform_attn_forward'])
+    _C = ext_loader.load_ext(
+        '_ext', ['ms_deform_attn_backward', 'ms_deform_attn_forward'])
 except:
     print("Failed to load custom C++ ops. Running on CPU mode Only!")
 

pre-requirements.txt

@@ -1,4 +1,3 @@
---extra-index-url https://download.pytorch.org/whl/cu113
-torch==1.12.1
-torchvision==0.13.1
-torchaudio==0.12.1
+torch
+torchvision
+torchaudio

requirements.txt

@@ -68,5 +68,5 @@ omegaconf
 git+https://github.com/facebookresearch/segment-anything.git
 git+https://github.com/IDEA-Research/GroundingDINO.git
 fairscale
---extra-index-url https://download.openmmlab.com/mmcv/dist/cu113/torch1.12/index.html
-mmcv==2.0.0rc4
+--extra-index-url https://download.openmmlab.com/mmcv/dist/cu117/torch2.0/index.html
+mmcv==2.0.0