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 Ye 26cae0fe03
Merge TorchScript Version (#76) 
from @yarkable
 		2021-03-05 11:31:55 +08:00
..
README.md Merge TorchScript Version (#76) 2021-03-05 11:31:55 +08:00
__init__.py Merge TorchScript Version (#76) 2021-03-05 11:31:55 +08:00
export_torchscript.py Merge TorchScript Version (#76) 2021-03-05 11:31:55 +08:00
modnet_torchscript.py Merge TorchScript Version (#76) 2021-03-05 11:31:55 +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