Browse Source

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

modifyDataloader
Glenn Jocher GitHub 3 years ago
parent
commit
9c31a66f5d
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      models/common.py

+ 1
- 1
models/common.py View File

@@ -277,7 +277,7 @@ class AutoShape(nn.Module):
# YOLOv5 input-robust model wrapper for passing cv2/np/PIL/torch inputs. Includes preprocessing, inference and NMS
conf = 0.25 # NMS confidence threshold
iou = 0.45 # NMS IoU threshold
classes = None # (optional list) filter by class
classes = None # (optional list) filter by class, i.e. = [0, 15, 16] for COCO persons, cats and dogs
multi_label = False # NMS multiple labels per box
max_det = 1000 # maximum number of detections per image


Loading…
Cancel
Save