mazhar004 / modnet-bgremover

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

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
Before removing the background from video	After replacing the background with new image in this video

Web View

Before removing the background	After removing the background

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

• Weights Detail

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

• A Trimap-Free Solution for Portrait Matting in Real Time under Changing Scenes
• Sample Female photo by Michael Dam on Unsplash
• Sample Male photo by Erik Lucatero on Unsplash