quoccuongle/grounding-dino
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
rentainhe
2023-05-02 19:41:34 +08:00
parent 0dc5ece5a2
commit a4dcf5d411
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
groundingdino/util/inference.py
View File

@@ -154,7 +154,7 @@ class Model:
caption=caption, caption=caption,
box_threshold=box_threshold, box_threshold=box_threshold,
text_threshold=text_threshold, text_threshold=text_threshold,
device=self.model.device) device=self.device)
source_h, source_w, _ = image.shape source_h, source_w, _ = image.shape
detections = Model.post_process_result( detections = Model.post_process_result(
source_h=source_h, source_h=source_h,
@@ -197,7 +197,7 @@ class Model:
caption=caption, caption=caption,
box_threshold=box_threshold, box_threshold=box_threshold,
text_threshold=text_threshold, text_threshold=text_threshold,
device=self.model.device) device=self.device)
source_h, source_w, _ = image.shape source_h, source_w, _ = image.shape
detections = Model.post_process_result( detections = Model.post_process_result(
source_h=source_h, source_h=source_h,