support gsam2 image predictor model
This commit is contained in:
8
grounding_dino/docker_test.py
Normal file
8
grounding_dino/docker_test.py
Normal 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!')
|
Reference in New Issue
Block a user