/modnet

A deep learning approach to remove background & adding new background image

Primary LanguagePython

MODNet Background Remover

Application

A deep learning approach to remove background and adding new background image

  • Remove background from images,videos & live webcam
  • Adding new background to those images,videos & webcam footage

Demo

Before removing the background After replacing the background with new image
Male.jpg Male.png
Before removing the background from video After replacing the background with new image in this video
Video

Web View

Before removing the background After removing the background
Female.jpeg Female.png

Installation

Python Version

  • Python == 3.8

Virtual Environment

Windows

  • python -m venv venv
  • .\venv\Scripts\activate
  • If any problem for scripts activation
    • Execute following command in administration mode
      • Set-ExecutionPolicy Unrestricted -Force
    • Later you can revert the change
      • Set-ExecutionPolicy restricted -Force

Linux

  • python -m venv venv
  • source venv/bin/activate

Library Installation

  • Library Install
    • pip install --upgrade pip
    • pip install --upgrade setuptools
    • pip install -r requirements.txt
    • To run in web interface
      • pip install -r web_requirements.txt

Pretrained Weights Download

Inference

Image

Single image

It will generate the output file in output/ folder

  • python inference.py --image image_path [Without background image]
  • python inference.py --image image_path --background True [With background image]
  • Example:
    • python inference.py --image assets/sample_image/female.jpeg
    • python inference.py --image assets/sample_image/male.jpeg --background True

Folder of images

It will generate the output file in output/ folder

  • python inference.py --folder folder_path [Without background image]
  • python inference.py --folder folder_path --background True [With background image]
  • Example:
    • python inference.py --folder assets/sample_image/
    • python inference.py --folder assets/sample_image/ --background True

Video

It will generate the output file in output/ folder

  • python inference.py --video video_path [Without background image]
  • python inference.py --video video_path --background True [With background image]
  • Example:
    • python inference.py --video assets/sample_video/sample.mp4
    • python inference.py --video assets/sample_video/sample.mp4 --background True

Webcam

  • python inference.py --webcam True [Without background image]
  • python inference.py --webcam True --background True [With background image]

Webinterface

  • python api.py
  • Click on this link/localhost
  • Upload the image and wait

Reference