From 9c31a66f5dfdf5b7f465ef20a470ad7e2a199354 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 25 Oct 2021 14:03:25 +0200
Subject: [PATCH] Update `AutoShape.forward()` model.classes example (#5324)

---
 models/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/common.py b/models/common.py
index f7f35c0..5da3569 100644
--- a/models/common.py
+++ b/models/common.py
@@ -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