Rimuovere il background dalle immagini con Python e rembg

rembg è un software scritto in Python per rimuovere il background dalle immagini.

Può essere usato sia in modalità standalone da riga di comando, che come libreria.

In questo articolo vedremo come usarlo come libreria.

Per installarla possiamo usare pip:

pip install rembg

Se volete usare la GPU:

pip install rembg[gpu]

Questo un primo esempio:

from rembg import remove

input_path = '/home/fermat/TEST/Apple wood.jpg'
output_path = '/home/fermat/TEST/output.png'

with open(input_path, 'rb') as i:
    with open(output_path, 'wb') as o:
        input = i.read()
        output = remove(input)
        o.write(output)

Se volessimo iterare su una cartella:

from pathlib import Path
from rembg import remove, new_session

session = new_session()

for file in Path('/home/fermat/TEST').glob('*.png'):
    input_path = str(file)
    output_path = str(file.parent / (file.stem + ".out.png"))

    with open(input_path, 'rb') as i:
        with open(output_path, 'wb') as o:
            input = i.read()
            output = remove(input, session=session)
            o.write(output)

Enjoy!