GroundingDINO source for verification 2026-05-22

This commit is contained in:
2026-05-22 16:45:02 +08:00
commit 9f82b61f46
68 changed files with 12625 additions and 0 deletions

8
docker_test.py Normal file
View File

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