Изображение и видео в фоновом режиме

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

окрестности

  • windows 10 64-bit
  • python 3.8.11
  • backgroundremover 0.1.9

предисловие

backgroundremoverЭто бесплатный инструмент командной строки с открытым исходным кодом, специально разработанный для удаления фона с изображений и видео.pythonЯзык написания, адрес проектаGitHub.com/Nadalobvious/80%….

Установка программы

можно использоватьpipУстановить

pip install backgroundremover

Конечно, также можно скачать исходный код для установки.

git clone https://github.com/nadermx/backgroundremover.git
cd backgroundremover
python setup.py install

На данный момент последней версией является 0.1.9, и последующие тесты также основаны на этой версии.

Далее необходимо установитьffmpeg, который используется для обработки видео. Скачайте скомпилированные бинарники с официального сайтаffmpeg.org/download.Контракты…, попробуйте загрузить более новую версию и добавить ее путь в системную переменную окружения

ffmpeg

Установите наконецtorch, потому что машина имеетgpu, установленgpuверсия, требуется версия 1.7 или выше

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

инструкции

Подготовьте исходное изображение с фоном

original

Выполните следующую команду, чтобы удалить фон, операция очень проста

backgroundremover -i original.jpg -o output.png

во время операцииgoogle driverдля загрузки необходимогоu2netмодель, но доступ к ней недоступен из-за внутренней сетевой среды

backgroundremover

В это время вы можете скачать его по ссылке ниже

u2net.pth u2net_human_seg.pth

Кроме того, также предоставляется адрес сетевого диска

Ссылка на сайт:disk.baidu.com/is/1ZY на V0sy E…Код извлечения:srix

Скачав файл модели, поместите его в~/.u2netПод папкой, если папки нет, создайте ее сами

u2net

Когда вы будете готовы, повторите приведенную выше команду, чтобы удалить заднюю часть, и вы сможете получить изображение с удаленным фоном.

backgroundremover

backgroundremoverДля удаления обратной стороны изображений и видео можно использовать три модели, а именно:u2netp,u2netиu2net_human_seg

Например, чтобы использоватьu2net_human_seg,можно использовать-mпараметр

backgroundremover -i original.jpg -m u2net_human_seg -o output.png

Если вы хотите обработать видео, вы можете сделать это

backgroundremover -i test.mp4 -tv -o output.mov

Фон видео удален и наложен на другое видео

backgroundremover -i test.mp4 -tov -tv videtobeoverlayed.mp4 -o output.mov

видео, созданное за кулисамиgifгифка

backgroundremover -i test.mp4 -tg -o output.gif

Дополнительные сведения об использовании сцены и настройках параметров см. в официальной документации.

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