Original Repo

Pre-trained model

create a directory named "pretrained"

Pre-trained model can be downloaded using following command:

$ gdown --id 1mcr7ALciuAsHCpLnrtG_eop5-EYhbCmz -O pretrained/modnet_photographic_portrait_matting.ckpt

or use the following link to download directly from google drive:

$ pip install -r requirements.txt

Note: Demo Tested with torch==1.1.0 and torchvision==0.3.0

python inference.py --input example.jpg