картинка на фон

глубокое обучение

Программно-аппаратная среда

  • windows 10 64bit
  • pytorch 1.7
  • rembg

Введение

rembgэто средство для удаления фонаpythonсредства не только вpythonОн вызывается и также предоставляет исполняемые программы, что очень удобно.

Установить

непосредственно черезpipУстановить

pip install rembg

или изgithubПотяните исходный код для установки

git clone https://github.com/danielgatis/rembg.git
cd rembg
pip install -r requirements.txt

После завершения установки будет сгенерированоrembgИсполняемый файл, вы можете использовать его для тестирования позже

Использование командной строки

Сначала скачайте файл модели, вот 2 адреса:

После загрузки файла скопируйте его в~/.u2netПод папкой этой папки по умолчанию нет, нужно создать самому

Далее вы можете использовать локальные образы для тестирования

rembg -o output_test.jpg test.jpg

-oэто обработанное изображение

rembg

rembg

rembg

rembg

Помимо обработки одного изображения,rembgОн также поддерживает обработку папок.Здесь поместите 2 вышеуказанные картинки в папку, а затем используйте команду

rembg -p test

rembg

rembgтакже обеспечиваетwebбоковой интерфейс, который выполняетrembg-serverЗапустите фоновую службу, которая используется здесьflaskРамка

rembg

Затем введите в браузере

http://127.0.0.1:5000/?url=https://image.xugaoxiang.com/imgs/2020/12/bf5f5553a596319b.jpg

url=Ниже приведен адрес изображения, вы можете увидетьrembgда поддержкаhttpsиз

rembg

Вызов rembg в коде

см. пример ниже

# app.py
import sys
from rembg.bg import remove

sys.stdout.buffer.write(remove(sys.stdin.buffer.read()))

а затем выполнить в терминале

cat bryant.jpg | python app.py > output.png

Результат запуска кода такой же, как и выше

использованная литература

Категории