Importante

Você está vendo a versão anterior da nova experiência da Alura que estamos preparando para você. Em breve, ela ganha uma identidade visual novinha totalmente pensada em potencializar seus estudos!

1
resposta

Usei exatamente o mesmo código da aula pra tirar a imagem de fundo mais simplesmente não funciona ?

Usei exatamente o mesmo código da aula pra tirar a imagem de fundo mais simplesmente não funciona ?

1 resposta

Oii, José. tudo bem?

Como sua mensagem não diz o que acontece ao executar (se aparece um erro ou se a imagem sai sem alteração), vou listar as causas mais comuns desse problema com o modelo briaai/RMBG-1.4 no Colab e como resolver cada uma.

1. Erro ao carregar o modelo (incompatibilidade de versão da biblioteca)

Esse é o motivo mais frequente hoje. O RMBG-1.4 usa código próprio (trust_remote_code=True), e versões mais recentes da biblioteca transformers, que o Colab instala por padrão, podem ser incompatíveis com esse código. Nesse caso, o erro costuma aparecer na linha do pipeline(...), geralmente com um AttributeError ou uma mensagem sobre o modelo.

Para resolver, instale uma versão anterior da biblioteca numa célula antes de tudo:

!pip install -q transformers==4.44.2

Depois, vá em Ambiente de execução > Reiniciar sessão e execute as células novamente. Sem reiniciar, o Colab continua usando a versão que já estava carregada.

2. Caminho da imagem incorreto

Confira se o image_path aponta para a sua imagem, e não para a da instrutora. No painel de arquivos à esquerda, clique nos três pontos ao lado da imagem, escolha Copiar caminho e cole entre aspas:

image_path = "/content/nome-da-sua-imagem.jpg"

Lembre também que os arquivos enviados ao Colab são apagados quando a sessão reinicia. Se você reiniciou o ambiente, faça o upload da imagem de novo.

3. O código roda, mas a imagem não aparece

A célula da aula só gera as variáveis e não mostra nada na tela. Para ver o resultado, execute em células separadas:

pillow_mask
pillow_image

4. Erro ao salvar a imagem

A imagem que você envia pode ser JPG, como na aula, sem problema. Mas o resultado gerado pelo modelo tem fundo transparente, e o formato JPG não aceita transparência. Se você tentou salvar o resultado com .jpg, vai receber um erro. Nesse caso, salve em PNG:

pillow_image.save("produto_sem_fundo.png")

Se nenhuma dessas soluções resolver, copie aqui a mensagem de erro completa que aparece no Colab (ou um print da célula) para identificarmos o problema com precisão.

Alura Conte com o apoio da comunidade Alura na sua jornada. Abraços e bons estudos!