From a4dcf5d411366d80b36c853372753330433a871a Mon Sep 17 00:00:00 2001
From: rentainhe <596106517@qq.com>
Date: Tue, 2 May 2023 19:41:34 +0800
Subject: [PATCH] fix bug
---
groundingdino/util/inference.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/groundingdino/util/inference.py b/groundingdino/util/inference.py
index 60e038d..f4c6c92 100644
--- a/groundingdino/util/inference.py
+++ b/groundingdino/util/inference.py
@@ -154,7 +154,7 @@ class Model:
caption=caption,
box_threshold=box_threshold,
text_threshold=text_threshold,
- device=self.model.device)
+ device=self.device)
source_h, source_w, _ = image.shape
detections = Model.post_process_result(
source_h=source_h,
@@ -197,7 +197,7 @@ class Model:
caption=caption,
box_threshold=box_threshold,
text_threshold=text_threshold,
- device=self.model.device)
+ device=self.device)
source_h, source_w, _ = image.shape
detections = Model.post_process_result(
source_h=source_h,