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

use model.device when calling legacy predict

pull/77/head
Darshat Shah 2023-04-27 12:15:11 +05:30
parent 498048b1b2
commit ff94310921
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
groundingdino/util/inference.py
View File

@ -153,7 +153,8 @@ class Model:
image=processed_image,
caption=caption,
box_threshold=box_threshold,
text_threshold=text_threshold)
text_threshold=text_threshold,
device=self.model.device)
source_h, source_w, _ = image.shape
detections = Model.post_process_result(
source_h=source_h,
@ -195,7 +196,8 @@ class Model:
image=processed_image,
caption=caption,
box_threshold=box_threshold,
text_threshold=text_threshold)
text_threshold=text_threshold,
device=self.model.device)
source_h, source_w, _ = image.shape
detections = Model.post_process_result(
source_h=source_h,