Added Dockerfile along with a file to test Docker (#307)

This commit is contained in:
Rohan Manzoor
2024-03-11 13:41:39 +05:00
committed by GitHub
parent d13643262e
commit c023468faf
2 changed files with 43 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!')