Hugging Face's logo

Spaces:
fmsfm
/
PictureLinear
Runtime error

App Files Community
PictureLinear / MODNet /torchscript /README.md
fmsfm's picture
fmsfm
Upload 37 files
555da6f
preview code
|
raw
history blame contribute delete
No virus
1.1 kB

A newer version of the Gradio SDK is available: 4.37.2

Upgrade

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