updates
This commit is contained in:
parent
3d1db0b5ac
commit
723431d6c3
|
@ -513,7 +513,7 @@ def non_max_suppression(prediction, conf_thres=0.5, iou_thres=0.5, multi_cls=Tru
|
|||
|
||||
method = 'vision_batch'
|
||||
nc = prediction[0].shape[1] - 5 # number of classes
|
||||
multi_cls = multi_cls and (nc > 1)
|
||||
multi_cls = multi_cls and (nc > 1) # allow multiple classes per anchor
|
||||
output = [None] * len(prediction)
|
||||
for image_i, pred in enumerate(prediction):
|
||||
# Apply conf constraint
|
||||
|
|
Loading…
Reference in New Issue