From 0c20a1d18d3ee4dfd50d6e1c00fc65f4d7da1076 Mon Sep 17 00:00:00 2001 From: Laughing <61612323+Laughing-q@users.noreply.github.com> Date: Tue, 25 Jul 2023 21:54:07 +0800 Subject: [PATCH] Fix `imgsz` in docs to integer to train, val (#3941) --- docs/modes/predict.md | 1 + docs/modes/train.md | 2 +- docs/modes/val.md | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/docs/modes/predict.md b/docs/modes/predict.md index bd3d55c..0602aa9 100644 --- a/docs/modes/predict.md +++ b/docs/modes/predict.md @@ -295,6 +295,7 @@ All supported arguments: | `source` | `'ultralytics/assets'` | source directory for images or videos | | `conf` | `0.25` | object confidence threshold for detection | | `iou` | `0.7` | intersection over union (IoU) threshold for NMS | +| `imgsz` | `640` | image size as scalar or (h, w) list, i.e. (640, 480) | | `half` | `False` | use half precision (FP16) | | `device` | `None` | device to run on, i.e. cuda device=0/1/2/3 or device=cpu | | `show` | `False` | show results if possible | diff --git a/docs/modes/train.md b/docs/modes/train.md index d853495..f9d844a 100644 --- a/docs/modes/train.md +++ b/docs/modes/train.md @@ -143,7 +143,7 @@ Training settings for YOLO models refer to the various hyperparameters and confi | `epochs` | `100` | number of epochs to train for | | `patience` | `50` | epochs to wait for no observable improvement for early stopping of training | | `batch` | `16` | number of images per batch (-1 for AutoBatch) | -| `imgsz` | `640` | size of input images as integer or w,h | +| `imgsz` | `640` | size of input images as integer | | `save` | `True` | save train checkpoints and predict results | | `save_period` | `-1` | Save checkpoint every x epochs (disabled if < 1) | | `cache` | `False` | True/ram, disk or False. Use cache for data loading | diff --git a/docs/modes/val.md b/docs/modes/val.md index cd8fd5f..7677267 100644 --- a/docs/modes/val.md +++ b/docs/modes/val.md @@ -48,7 +48,7 @@ Validation settings for YOLO models refer to the various hyperparameters and con | Key | Value | Description | |---------------|---------|--------------------------------------------------------------------| | `data` | `None` | path to data file, i.e. coco128.yaml | -| `imgsz` | `640` | image size as scalar or (h, w) list, i.e. (640, 480) | +| `imgsz` | `640` | size of input images as integer | | `batch` | `16` | number of images per batch (-1 for AutoBatch) | | `save_json` | `False` | save results to JSON file | | `save_hybrid` | `False` | save hybrid version of labels (labels + additional predictions) |