mirrors
/
GroundingDINO
mirror of https://github.com/IDEA-Research/GroundingDINO.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge 919c695a10 into 856dde20ae

pull/361/merge
Egor Korovko 2024-10-08 04:02:10 +03:00 committed by GitHub
parent 856dde20ae 919c695a10
commit 0af8aa9c20
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docker_test.py
View File

@ -2,7 +2,7 @@ from groundingdino.util.inference import load_model, load_image, predict, annota
import torch
import cv2
model = load_model("groundingdino/config/GroundingDINO_SwinT_OGC.pyy", "weights/groundingdino_swint_ogc.pth")
model = load_model("groundingdino/config/GroundingDINO_SwinT_OGC.py", "weights/groundingdino_swint_ogc.pth")
model = model.to('cuda:0')
print(torch.cuda.is_available())
print('DONE!')
print('DONE!')