chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-05 02:20:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
Zhijie Lin 2023-05-25 17:23:32 +08:00
parent b5bbd4a2f2
commit c33e75d3a3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
GroundingModel.py
View File

@ -54,7 +54,7 @@ class GroundingModule(nn.Module):
prompt = prompt + "."
_, image_tensor = image_transform_grounding(original_image)
boxes, logits, phrases = predict(self.grounding_model,
image_tensor, prompt, box_threshold, text_threshold, device=self.device)
image_tensor, prompt, box_threshold, text_threshold, device='cpu')
print(phrases)
# from PIL import Image, ImageDraw, ImageFont
H, W = original_image.size[1], original_image.size[0]