Dagoland: Rembg: Uma ferramenta para remover imagens de fundo

Rembg é uma ferramenta para remover imagens de fundo.

Requisitos
python: >3.7, <3.11

INSTALAÇÃO

sudo apt install python3-pip

Suporte da CPU:

pip install rembg

Suporte GPU:

pip install rembg[gpu]

Uso como um cli

Após a etapa de instalação, você pode usar o rembg apenas digitando rembg na janela do seu terminal.

O rembgcomando possui 3 subcomandos, um para cada tipo de entrada:

i para arquivos
p para pastas
s para servidor http

Você pode obter ajuda sobre o comando principal usando:

rembg --help

Além disso, sobre todos os subcomandos usando:

rembg <COMANDO> --help

rembg i

Usado quando a entrada e a saída são arquivos.

Remover o fundo de uma imagem remota

curl -s http://input.png | rembg i > output.png

Remova o plano de fundo de um arquivo local

rembg i path/to/input.png path/to/output.png

Remova o plano de fundo especificando um modelo

rembg i -m u2netp path/to/input.png path/to/output.png

Remova o fundo retornando apenas a máscara

rembg i -om path/to/input.png path/to/output.png

Remova o fundo aplicando um alpha matting

rembg i -a path/to/input.png path/to/output.png

rembg p

Usado quando a entrada e a saída são pastas.

Remova o plano de fundo de todas as imagens em uma pasta

rembg p path/to/input path/to/output

O mesmo de antes, mas observando os arquivos novos/alterados para processar

rembg p -w path/to/input path/to/output

rembg s

Usado para iniciar o servidor http.

Para ver a documentação completa dos endpoints, acesse: http://localhost:5000/docs.

Remova o plano de fundo de um URL de imagem

curl -s "http://localhost:5000/?url=http://input.png" -o output.png

Remover o plano de fundo de uma imagem carregada

curl -s -F file=@/path/to/input.jpg "http://localhost:5000" -o output.png

Alguns tutoriais em vídeo:
https://www.youtube.com/watch?v=3xqwpXjxyMQ
https://www.youtube.com/watch?v=dFKRGXdkGJU
https://www.youtube.com/watch?v=Ai-BS_T7yjE
https://www.youtube.com/watch?v=dFKRGXdkGJU
https://www.youtube.com/watch?v=D7W-C0urVcQ

Referências

https://github.com/danielgatis/rembg
https://arxiv.org/pdf/2005.09007.pdf
https://github.com/NathanUA/U-2-Net
https://github.com/pymatting/pymatting