Skip to content

Commit 0716131

Browse files
committed
Update AutoShape.forward() model.classes example
1 parent ca19df5 commit 0716131

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -277,7 +277,7 @@ class AutoShape(nn.Module):
277277
# YOLOv5 input-robust model wrapper for passing cv2/np/PIL/torch inputs. Includes preprocessing, inference and NMS
278278
conf = 0.25 # NMS confidence threshold
279279
iou = 0.45 # NMS IoU threshold
280-
classes = None # (optional list) filter by class
280+
classes = None # (optional list) filter by class, i.e. = [0, 15, 16] for COCO persons, cats and dogs
281281
multi_label = False # NMS multiple labels per box
282282
max_det = 1000 # maximum number of detections per image
283283

0 commit comments

Comments
 (0)