Photoshop не нужен: Python удаляет фон с 1000+ фото за несколько минут

Мы можем заменить часы ручной работы двумя строчками кода на Python с помощью библиотеки rembg. Она использует нейросети для точного "вырезания" объектов и хорошо подходит для быстрой обработки большого количества фотографий

Нам понадобится сама библиотека и Pillow для работы с графическими объектами:

pip install rembg pillow

Импортируем

from rembg import remove
from PIL import Image

Определяем входной и выходной файл 👇

input_file = 'source_photo.jpg'
output_file = 'result_transparent.png'

input_image = Image.open(input_file)

🟡 Вся суть процесса здесь. Функция remove() автоматически обрабатывает изображение и возвращает его с прозрачным фоном. Сохраняем результат обязательно в PNG, чтобы сохранить прозрачность

output_image = remove(input_image)

output_image.save(output_file)

Готово =) Можно использовать его для автоматизации вырезания картинок для маркетплейсов, подготовки контента или наполнения своих пет-проектов ⚡️

🚀 Пост Guru Python: @sergey_filichkin_blog