Fix AutoAugment abs translate calc

rwightman · rwightman · commit 4002c0d4ce4d · 2019-09-01T22:07:45.000-07:00
diff --git a/timm/data/transforms.py b/timm/data/transforms.py
@@ -179,8 +179,12 @@ def transforms_imagenet_train(
         transforms.RandomHorizontalFlip()
     ]
     if auto_augment:
+        if isinstance(img_size, tuple):
+            img_size_min = min(img_size)
+        else:
+            img_size_min = img_size
         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]),
         )
         if interpolation and interpolation != 'random':

Original file line number	Diff line number	Diff line change
`@@ -179,8 +179,12 @@ def transforms_imagenet_train(`
`179`	`179`	`transforms.RandomHorizontalFlip()`
`180`	`180`	`]`
`181`	`181`	`if auto_augment:`
	`182`	`+ if isinstance(img_size, tuple):`
	`183`	`+ img_size_min = min(img_size)`
	`184`	`+ else:`
	`185`	`+ img_size_min = img_size`
`182`	`186`	`aa_params = dict(`
`183`		`- translate_const=img_size[-1] // 2 - 1,`
	`187`	`+ translate_const=int(img_size_min * 0.45),`
`184`	`188`	`img_mean=tuple([min(255, round(255 * x)) for x in mean]),`
`185`	`189`	`)`
`186`	`190`	`if interpolation and interpolation != 'random':`