Skip to content

Commit 429f9cb

Browse files
authored
fixed predict threh set (#3679)
1 parent 2e40aac commit 429f9cb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddlex/inference/models/object_detection/predictor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ def process(
245245
boxes = self.post_op(
246246
preds_list,
247247
datas,
248-
threshold=threshold or self.threshold,
248+
threshold=threshold if threshold is not None else self.threshold,
249249
layout_nms=layout_nms or self.layout_nms,
250250
layout_unclip_ratio=layout_unclip_ratio or self.layout_unclip_ratio,
251251
layout_merge_bboxes_mode=layout_merge_bboxes_mode

0 commit comments

Comments
 (0)