Switch to PIL backend

datumbox · datumbox · commit 8b530360cd5b · 2022-11-17T12:09:12.000Z
diff --git a/references/classification/presets.py b/references/classification/presets.py
@@ -18,7 +18,6 @@ def __init__(
         random_erase_prob=0.0,
     ):
         trans = [
-            transforms.ToImageTensor(),
             transforms.RandomResizedCrop(crop_size, interpolation=interpolation, antialias=True),
         ]
         if hflip_prob > 0:
@@ -35,6 +34,7 @@ def __init__(
                 trans.append(transforms.AutoAugment(policy=aa_policy, interpolation=interpolation))
         trans.extend(
             [
+                transforms.PILToTensor(),
                 transforms.ConvertImageDtype(torch.float),
                 transforms.Normalize(mean=mean, std=std),
             ]
@@ -61,9 +61,9 @@ def __init__(
 
         self.transforms = transforms.Compose(
             [
-                transforms.ToImageTensor(),
                 transforms.Resize(resize_size, interpolation=interpolation, antialias=True),
                 transforms.CenterCrop(crop_size),
+                transforms.PILToTensor(),
                 transforms.ConvertImageDtype(torch.float),
                 transforms.Normalize(mean=mean, std=std),
             ]

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,6 @@ def __init__(`
`18`	`18`	`random_erase_prob=0.0,`
`19`	`19`	`):`
`20`	`20`	`trans = [`
`21`		`- transforms.ToImageTensor(),`
`22`	`21`	`transforms.RandomResizedCrop(crop_size, interpolation=interpolation, antialias=True),`
`23`	`22`	`]`
`24`	`23`	`if hflip_prob > 0:`
`@@ -35,6 +34,7 @@ def __init__(`
`35`	`34`	`trans.append(transforms.AutoAugment(policy=aa_policy, interpolation=interpolation))`
`36`	`35`	`trans.extend(`
`37`	`36`	`[`
	`37`	`+ transforms.PILToTensor(),`
`38`	`38`	`transforms.ConvertImageDtype(torch.float),`
`39`	`39`	`transforms.Normalize(mean=mean, std=std),`
`40`	`40`	`]`
`@@ -61,9 +61,9 @@ def __init__(`
`61`	`61`
`62`	`62`	`self.transforms = transforms.Compose(`
`63`	`63`	`[`
`64`		`- transforms.ToImageTensor(),`
`65`	`64`	`transforms.Resize(resize_size, interpolation=interpolation, antialias=True),`
`66`	`65`	`transforms.CenterCrop(crop_size),`
	`66`	`+ transforms.PILToTensor(),`
`67`	`67`	`transforms.ConvertImageDtype(torch.float),`
`68`	`68`	`transforms.Normalize(mean=mean, std=std),`
`69`	`69`	`]`