examples icon indicating copy to clipboard operation
examples copied to clipboard

Neural Style fails if style image has an alpha channel

Open hackf5 opened this issue 5 years ago • 0 comments

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.

hackf5 avatar Mar 20 '20 17:03 hackf5