glenn-jocher
commited on
Commit
•
9c31a66
1
Parent(s):
ca19df5
Update `AutoShape.forward()` model.classes example (#5324)
Browse files- models/common.py +1 -1
models/common.py
CHANGED
@@ -277,7 +277,7 @@ class AutoShape(nn.Module):
|
|
277 |
# YOLOv5 input-robust model wrapper for passing cv2/np/PIL/torch inputs. Includes preprocessing, inference and NMS
|
278 |
conf = 0.25 # NMS confidence threshold
|
279 |
iou = 0.45 # NMS IoU threshold
|
280 |
-
classes = None # (optional list) filter by class
|
281 |
multi_label = False # NMS multiple labels per box
|
282 |
max_det = 1000 # maximum number of detections per image
|
283 |
|
|
|
277 |
# YOLOv5 input-robust model wrapper for passing cv2/np/PIL/torch inputs. Includes preprocessing, inference and NMS
|
278 |
conf = 0.25 # NMS confidence threshold
|
279 |
iou = 0.45 # NMS IoU threshold
|
280 |
+
classes = None # (optional list) filter by class, i.e. = [0, 15, 16] for COCO persons, cats and dogs
|
281 |
multi_label = False # NMS multiple labels per box
|
282 |
max_det = 1000 # maximum number of detections per image
|
283 |
|