Unverified 提交 9c31a66f authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

上级 ca19df5f
...@@ -277,7 +277,7 @@ class AutoShape(nn.Module): ...@@ -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 # YOLOv5 input-robust model wrapper for passing cv2/np/PIL/torch inputs. Includes preprocessing, inference and NMS
conf = 0.25 # NMS confidence threshold conf = 0.25 # NMS confidence threshold
iou = 0.45 # NMS IoU 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 multi_label = False # NMS multiple labels per box
max_det = 1000 # maximum number of detections per image max_det = 1000 # maximum number of detections per image
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论