Query Regarding Transforms used for input image and style image about fast-neural-style

[abhiskk]

This line converts the image to 0-1 range. I would suggest running the code or reading the documentation for such doubts. Opening an issue for these doubts is not appropriate.

[ssundar6087]

@abhiskk I apologize if this came across as a trivial question to you. My question was specifically regarding the training script.

There you use the pytorch provided dataset.ImageFolder as follows :
transform = transforms.Compose([
transforms.Resize(args.image_size),
transforms.CenterCrop(args.image_size),
transforms.ToTensor(),
transforms.Lambda(lambda x: x.mul(255))
])
train_dataset = datasets.ImageFolder(args.dataset, transform)
train_loader = DataLoader(train_dataset, batch_size=args.batch_size)

From my limited understanding, the ImageFolder function doesn't normalize the input to floats in the range of 0-1. The other transforms such as resize, centercrop and ToTensor shouldn't change the data range either and hence my thought was that you don't need the x.mul(255) on the input. If I am mistaken, please correct me.

[abhiskk]

Yes, transforms.ToTensor() converts the image to 0-1 range hence the multiply was needed.

[ssundar6087]

Thanks for clarifying. Apologies once again.

[abhiskk]

No worries, happy to help 👍