pytorch/examples

Neural Style fails if style image has an alpha channel

Open

#738 ouverte le 20 mars 2020

Voir sur GitHub
 (0 commentaires) (0 réactions) (0 assignés)Python (21 634 stars) (9 429 forks)batch import
help wanted

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.

Guide contributeur

Stack technique
python
Domaine
machine learning
Type d'issue
bug
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
2
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
under 1 hour
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
fresh
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
clear
Prérequis
PythonPIL/PillowImage handling
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
80
Direction de recherche
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.