Change-Background

Change background of image using semantic segmentation

Click and change your background now

Running script

Weights

Download weights from here:

https://drive.google.com/file/d/1VNYZ1X5bnIZFIVIgNi9JwqCNIvioxexe/view?usp=sharing

Run

python main.py --image --bg_image --weights --debug --output_dir

Input

Output

Example

python main.py --image demo_images/image1.jpg --bg_image demo_images/background.jpg --weights weights/hrnetv2_hrnet18_person_dataset_120.pth --debug