hi, i’m working with rgb images of diabetic retinopathy dataset. since i want to train my unet model, i have to do transform first, but i confuse about which transform should i use. i’ve read the documentation of transform ToTensor and it says that it’ll return [0,1] image, so it’ll be binary input to the model? which one should i use? thanks!
transforms.ToTensor() will return a normalized tensors with values in the range
transforms.PILToTensor() will return a tensor using the same
dtype and value range as the original input image.
ah get it! which one is better to use for my problem?
The more common approach is to use
ToTensor to create normalized
FloatTensors as you would need to transform the
dtype to e.g.
float32 manually otherwise.