glenn-jocher commited on
Commit
9c31a66
1 Parent(s): ca19df5

Update `AutoShape.forward()` model.classes example (#5324)

Browse files
Files changed (1) hide show
  1. 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