ImageNet preprocessing

[pytorch]

ImageNet preprocessing

[cbasavaraj]

Ok thanks! I thought Scale was for converting from [0, 255] to [0, 1]. But I see that that happens in transforms.ToTensor().

[soumith]

You can see in Line 100 and 110, that both train and val datasets are normalized.

https://github.com/pytorch/examples/blob/master/imagenet/main.py#L100-L110

[cbasavaraj]

Thanks, but maybe my question was not clear. Both are normalised, but train is not scaled.

[fmassa]

@cbasavaraj The Scale transform resizes the image such that it's smallest size is 256.