We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4d2e809 commit c048160Copy full SHA for c048160
1 file changed
patched_yolo_infer/nodes/CombineDetections.py
@@ -79,7 +79,7 @@ def __init__(
79
] # make str list
80
81
# Invoke the NMS:
82
- if self.class_agnostic_nms:
+ if not self.class_agnostic_nms:
83
self.filtered_indices = self.agnostic_nms(
84
torch.tensor(self.detected_cls_id_list_full),
85
torch.tensor(self.detected_conf_list_full),
0 commit comments