Skip to content

Latest commit

 

History

History
16 lines (13 loc) · 1.25 KB

About_training_size_cn.md

File metadata and controls

16 lines (13 loc) · 1.25 KB

训练尺寸说明

YOLOv6支持三种训练尺寸模式。

1. 正方形尺寸训练

如果只给 --img-size 指定一个数字,例如 --img-size 640,则图像的长边将被缩放到 640(保持长宽比),短边等比例缩放后,将被填充到 640。送入模型的图像的分辨率将变为(640, 640, 3)。

2. 矩形尺寸训练

如果传递了 --img-size 640--rect,则图像的长边将被缩放到 640(保持长宽比),短边将被等比例缩放,然后填充到 32 的倍数(如果需要)。 例如,如果一张图像的形状为(720, 1280, 3),在等比例缩放后,它的形状将变为(360, 640, 3),但是 360 不是 32 的倍数,因此它将被填充为(384, 640, 3)。

3. 特定尺寸

在矩形尺寸训练模式下,训练过程可能有不同的训练尺寸,例如(1080, 1920, 3)和(1200, 1600, 3)。如果您想指定一个尺寸,可以使用 --specific-shape 命令,并使用 --height--width 指定您的训练尺寸,例如:

python tools/train.py --data data/dataset.yaml --conf configs/yolov6n.py --specific-shape --width 1920 --height 1080

那么,无论数据集中图片的形状是什么,训练数据的分辨率将都是 (1080, 1920, 3)。