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 (9 429 forks)batch import
help wanted

Métriques du dépôt

Stars
 (21 634 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

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

Direction de recherche
Appliquez la correction en ajoutant `.convert('RGB')` au chargement de l'image de style à la ligne 55 de `neural style.py`. Assurez vous que la fonction de normalisation fonctionne avec des images à 3 canaux. Testez avec une image qui a un canal alpha.
Stack technique
pythonpytorch
Domaine
aimachine learningcomputer vision
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.
1
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
Moins d'une heure
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
GitPython
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
90