Update pipeline/tool/object_detetction_model.py

pipeline/tool/object_detetction_model.py

@@ -42,7 +42,7 @@ class GroundingDINO:
         image_source, image = load_image(IMAGE_PATH)
         if use_text_rec:
             # 在场景文本中下调boxthreshold
-            boxes, logits, phrases = predict(model=self.model,image=image,caption=content,box_threshold=0.2,text_threshold=self.TEXT_TRESHOLD)#,device='cuda:0')
+            boxes, logits, phrases = predict(model=self.model,image=image,caption=content,box_threshold=0.2,text_threshold=self.TEXT_TRESHOLD, device='cpu')
             h, w, _ = image_source.shape
             torch_boxes = boxes * torch.Tensor([w, h, w, h])
             xyxy = box_convert(boxes=torch_boxes, in_fmt="cxcywh", out_fmt="xyxy").numpy()
@@ -72,7 +72,7 @@ class GroundingDINO:
         else:
             new_path = new_path + IMAGE_PATH.split('/')[-1]
             print(content)
-            boxes, logits, phrases = predict(model=self.model,image=image,caption=content,box_threshold=self.BOX_TRESHOLD,text_threshold=self.TEXT_TRESHOLD)#,device='cuda:0')
+            boxes, logits, phrases = predict(model=self.model,image=image,caption=content,box_threshold=self.BOX_TRESHOLD,text_threshold=self.TEXT_TRESHOLD, device='cpu')
             annotated_frame = annotate(image_source=image_source, boxes=boxes, logits=logits, phrases=phrases)
             cv2.imwrite(new_path, annotated_frame)
             h, w, _ = image_source.shape