3.4

公用设施

图像压缩

将单个图像文件压缩至更小尺寸，同时保留其宽高比和质量。如果输入图像小于最大尺寸，则不会调整其大小。

from pathlib import Path

from ultralytics.data.utils import compress_one_image

for f in Path("path/to/dataset").rglob("*.jpg"):
    compress_one_image(f)  

自动分割数据集

自动将数据集分割成 train/val/test 分割，并将分割结果保存到 autosplit_*.txt 文件。该功能将使用随机取样，而使用 fraction 培训论据.

from ultralytics.data.utils import autosplit

autosplit(  
    path="path/to/images",
    weights=(0.9, 0.1, 0.0),  # (train, validation, test) fractional splits
    annotated_only=False,  # split only images with annotation file when True
)