mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix AutoAugment abs translate calc

Browse Source
This commit is contained in:
Ross Wightman 2019-09-01 22:07:45 -07:00
parent c06274e5a2
commit 4002c0d4ce
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
timm/data/transforms.py
View File

@ -179,8 +179,12 @@ def transforms_imagenet_train(
transforms.RandomHorizontalFlip() transforms.RandomHorizontalFlip()
] ]
if auto_augment: if auto_augment:
if isinstance(img_size, tuple):
img_size_min = min(img_size)
else:
img_size_min = img_size
aa_params = dict( aa_params = dict(
translate_const=img_size[-1] // 2 - 1, translate_const=int(img_size_min * 0.45),
img_mean=tuple([min(255, round(255 * x)) for x in mean]), img_mean=tuple([min(255, round(255 * x)) for x in mean]),
) )
if interpolation and interpolation != 'random': if interpolation and interpolation != 'random':