mirror
/
MODNet
mirror of https://github.com/ZHKKKe/MODNet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
MODNet/TorchScript
History
kevin e481792869 add torchscript 2021-03-05 14:19:30 +08:00
..
README.md add torchscript 2021-03-05 14:19:30 +08:00
__init__.py add torchscript 2021-03-05 14:19:30 +08:00
export_torchscript.py add torchscript 2021-03-05 14:19:30 +08:00
modnet_torchscript.py add torchscript 2021-03-05 14:19:30 +08:00

README.md

Usage:


python export_torchscript.py \
	--ckpt-path pretrained/modnet_photographic_portrait_matting.ckpt\
	--out-dir scripted_model

Official TorchScript model:

BaiduCloudDisk, extract_code=dm9e