Open
Description
In fast_neural_style/neural_style/neural_style.py
line 55, if the style image has an alpha channel, then the generated tensor has 4 dimensions and this causes utils.normalize_batch
to throw due to a tensor dimension mismatch a few lines down.
I've fixed this by appending .convert('RGB')
so line 55 now reads
style = utils.load_image(args.style_image, size=args.style_size).convert('RGB')
The ImageFolder
data loader does the same transformation, however, maybe a warning should be issued since it is the key file.