Skip to content

Latest commit

 

History

History

torchscript

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
__init__.py
__init__.py
 
 
export_torchscript.py
export_torchscript.py
 
 
modnet_torchscript.py
modnet_torchscript.py
 
 

README.md

MODNet - TorchScript Model

This TorchScript version of MODNet is provided by @yarkable from the community.
Please note that the PyTorch version required for this TorchScript export function is higher than the official MODNet code (torch>=1.2.0).

You can also download the TorchScript version of the official Image Matting Model from this link with the exextraction code dm9e.

To export the TorchScript version of MODNet (assuming you are currently in project root directory):

  1. Download the pre-trained Image Matting Model from this link and put the model into the folder MODNet/pretrained/.

  2. Ensure your PyTorch version >= 1.2.0.

  3. Export the TorchScript version of MODNet by:

    python -m torchscript.export_torchscript \
    --ckpt-path=pretrained/modnet_photographic_portrait_matting.ckpt \
    --output-path=pretrained/modnet_photographic_portrait_matting.torchscript