Fix `imgsz` in docs to integer to train, val (#3941)

single_channel
Laughing 1 year ago committed by GitHub
parent 9f5ab67ba2
commit 0c20a1d18d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -295,6 +295,7 @@ All supported arguments:
| `source` | `'ultralytics/assets'` | source directory for images or videos | | `source` | `'ultralytics/assets'` | source directory for images or videos |
| `conf` | `0.25` | object confidence threshold for detection | | `conf` | `0.25` | object confidence threshold for detection |
| `iou` | `0.7` | intersection over union (IoU) threshold for NMS | | `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) | | `half` | `False` | use half precision (FP16) |
| `device` | `None` | device to run on, i.e. cuda device=0/1/2/3 or device=cpu | | `device` | `None` | device to run on, i.e. cuda device=0/1/2/3 or device=cpu |
| `show` | `False` | show results if possible | | `show` | `False` | show results if possible |

@ -143,7 +143,7 @@ Training settings for YOLO models refer to the various hyperparameters and confi
| `epochs` | `100` | number of epochs to train for | | `epochs` | `100` | number of epochs to train for |
| `patience` | `50` | epochs to wait for no observable improvement for early stopping of training | | `patience` | `50` | epochs to wait for no observable improvement for early stopping of training |
| `batch` | `16` | number of images per batch (-1 for AutoBatch) | | `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` | `True` | save train checkpoints and predict results |
| `save_period` | `-1` | Save checkpoint every x epochs (disabled if < 1) | | `save_period` | `-1` | Save checkpoint every x epochs (disabled if < 1) |
| `cache` | `False` | True/ram, disk or False. Use cache for data loading | | `cache` | `False` | True/ram, disk or False. Use cache for data loading |

@ -48,7 +48,7 @@ Validation settings for YOLO models refer to the various hyperparameters and con
| Key | Value | Description | | Key | Value | Description |
|---------------|---------|--------------------------------------------------------------------| |---------------|---------|--------------------------------------------------------------------|
| `data` | `None` | path to data file, i.e. coco128.yaml | | `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) | | `batch` | `16` | number of images per batch (-1 for AutoBatch) |
| `save_json` | `False` | save results to JSON file | | `save_json` | `False` | save results to JSON file |
| `save_hybrid` | `False` | save hybrid version of labels (labels + additional predictions) | | `save_hybrid` | `False` | save hybrid version of labels (labels + additional predictions) |

Loading…
Cancel
Save