Удаление фона изображения с помощью Python

Научитесь легко удалять фон изображения с помощью Python. Узнайте, как можно легко удалить фон изображений в Python. Овладейте искусством обработки изображений, чтобы создавать потрясающие визуальные эффекты и безупречный дизайн.

Для этого мы используем инструмент с открытым исходным кодом rembg, и нам нужно всего лишь несколько строк кода.

Установить рембг

Чтобы использовать его с процессором, используйте следующую команду:

pip install rembg

Если у вас есть графический процессор, вы можете использовать:

pip install rembg[gpu]

Убедитесь, что установлена ​​правильная версия Pillow. Должно быть >= 9.3.0

import PIL
PIL.__version__

Если это не так, вы можете попробовать использовать эту команду (мне пришлось использовать этот обходной путь при использовании Google Colab):

pip install --ignore-installed Pillow==9.3.0

Удалить фон

Теперь вы можете легко удалить фон следующим образом:

from rembg import remove

input_path = 'input.png'
output_path = '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)

Это загрузит изображение, применит удаление фона и запишет результат в новое выходное изображение.

Используйте его как CLI

Вы также можете использовать rembg из командной строки:

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