Unverified 提交 19c56e60 authored 作者: Diego Montes's avatar Diego Montes 提交者: GitHub

Fix `imgsz` bug (#5948)

* fix imgsz bug * Update detect.py Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 8f354362
...@@ -38,7 +38,7 @@ from utils.torch_utils import select_device, time_sync ...@@ -38,7 +38,7 @@ from utils.torch_utils import select_device, time_sync
@torch.no_grad() @torch.no_grad()
def run(weights=ROOT / 'yolov5s.pt', # model.pt path(s) def run(weights=ROOT / 'yolov5s.pt', # model.pt path(s)
source=ROOT / 'data/images', # file/dir/URL/glob, 0 for webcam source=ROOT / 'data/images', # file/dir/URL/glob, 0 for webcam
imgsz=640, # inference size (pixels) imgsz=(640, 640), # inference size (height, width)
conf_thres=0.25, # confidence threshold conf_thres=0.25, # confidence threshold
iou_thres=0.45, # NMS IOU threshold iou_thres=0.45, # NMS IOU threshold
max_det=1000, # maximum detections per image max_det=1000, # maximum detections per image
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论