Quellcode durchsuchen

Add NMS threshold checks (#3020)

* Add NMS threshold checks

* fix
modifyDataloader
Glenn Jocher GitHub vor 3 Jahren
Ursprung
Commit
41cc7caee6
Es konnte kein GPG-Schlüssel zu dieser Signatur gefunden werden GPG-Schlüssel-ID: 4AEE18F83AFDEB23
1 geänderte Dateien mit 4 neuen und 0 gelöschten Zeilen
  1. +4
    -0
      utils/general.py

+ 4
- 0
utils/general.py Datei anzeigen

@@ -476,6 +476,10 @@ def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=Non
nc = prediction.shape[2] - 5 # number of classes
xc = prediction[..., 4] > conf_thres # candidates

# Checks
assert 0 <= conf_thres <= 1, f'Invalid Confidence threshold {conf_thres}, valid values are between 0.0 and 1.0'
assert 0 <= iou_thres <= 1, f'Invalid IoU {iou_thres}, valid values are between 0.0 and 1.0'

# Settings
min_wh, max_wh = 2, 4096 # (pixels) minimum and maximum box width and height
max_det = 300 # maximum number of detections per image

Laden…
Abbrechen
Speichern