pytorch/examples

Neural Style fails if style image has an alpha channel

Open

Aperta il 20 mar 2020

Vedi su GitHub
 (0 commenti) (0 reazioni) (0 assegnatari)Python (21.634 star) (9429 fork)batch import
help wanted

Descrizione

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.

Guida contributor

Tech stack
python
Dominio
machine learning
Tipo issue
bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
under 1 hour
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
fresh
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
clear
Prerequisiti
PythonPIL/PillowImage handling
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
80
Direzione di ricerca
The fix is to modify line 55 in fast neural style/neural style/neural style.py by appending .convert('RGB') after loading the style image. Optionally, add a warning when alpha channel is discarded. The file is part of the pytorch/examples repository. No additional research needed.