Home Explore Help
Register Sign In
zhengchun
/
PaddleX
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

fixed predict threh set (#3678)

学卿 8 months ago
parent
48d7333596
commit
76b6ebcf40
1 changed files with 1 additions and 1 deletions
Split View Show Diff Stats
  1. 1 1
      paddlex/inference/models/object_detection/predictor.py

+ 1 - 1
paddlex/inference/models/object_detection/predictor.py
View File 

@@ -245,7 +245,7 @@ class DetPredictor(BasicPredictor):
 
         boxes = self.post_op(
 
             preds_list,
 
             datas,
 
-            threshold=threshold or self.threshold,
 
+            threshold=threshold if threshold is not None else self.threshold,
 
             layout_nms=layout_nms or self.layout_nms,
 
             layout_unclip_ratio=layout_unclip_ratio or self.layout_unclip_ratio,
 
             layout_merge_bboxes_mode=layout_merge_bboxes_mode