pytorch/examples

Neural Style fails if style image has an alpha channel

Open

#738 aberto em 20 de mar. de 2020

Ver no GitHub
 (0 comments) (0 reactions) (0 assignees)Python (9.429 forks)batch import
help wanted

Métricas do repositório

Stars
 (21.634 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

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.

Guia do colaborador

Direção de pesquisa
Aplique a correção adicionando `.convert('RGB')` ao carregamento da imagem de estilo na linha 55 do `neural style.py`. Certifique se de que a função de normalização funcione com imagens de 3 canais. Teste com uma imagem que tenha um canal alfa.
Pilha de tecnologia
pythonpytorch
Domain
aimachine learningcomputer vision
Tipo Issue
Bug
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
1
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
Menos de 1 hora
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
GitPython
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
90