HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Update pyproject.toml and Docs (#7274)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Yaofu <voipman@sina.cn>
Co-authored-by: Umit Kacar, PhD <kacarumit.phd@gmail.com>
This commit is contained in:
Glenn Jocher 2024-01-04 01:40:13 +01:00 • committed by GitHub
parent e687c09423
commit f702b34a50
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 18 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ultralytics/data/augment.py
View file

@ -1005,7 +1005,7 @@ def classify_transforms(
crop_fraction (float): fraction of image to crop. default is 1.0.
Returns:
T.Compose: torchvision transforms
(T.Compose): torchvision transforms
"""
if isinstance(size, (tuple, list)):
@ -1064,13 +1064,12 @@ def classify_augmentations(
hsv_h (float): image HSV-Hue augmentation (fraction)
hsv_s (float): image HSV-Saturation augmentation (fraction)
hsv_v (float): image HSV-Value augmentation (fraction)
contrast (float): image contrast augmentation (fraction)
force_color_jitter (bool): force to apply color jitter even if auto augment is enabled
erasing (float): probability of random erasing
interpolation (T.InterpolationMode): interpolation mode. default is T.InterpolationMode.BILINEAR.
Returns:
T.Compose: torchvision transforms
(T.Compose): torchvision transforms
"""
# Transforms to apply if albumentations not installed
if not isinstance(size, int):