torchscript_mobilenetv2_fp32.pth

DBraun / PyTorchTOP

GPU PyTorch TOP in TouchDesigner with CUDA-enabled OpenCV

Other

77 stars 5 forks source link

torchscript_mobilenetv2_fp32.pth #9

Closed ryansimrell closed 3 years ago

ryansimrell commented 3 years ago

After install the cplusplus_background_matting top gives an error saying it is missing this file... "PyTorchTop/models/torchscript_mobilenetv2_fp32.pth" anyone know why that wouldn't generate?

DBraun commented 3 years ago

The pretrained torchscript model files can be downloaded from here https://github.com/PeterL1n/BackgroundMattingV2#download. Place them in the models folder. Also be careful to check the Bytes per Model Input custom parameter on the plugin. It should be 2 for 16-bit models and 4 for 32-bit models.

ryansimrell commented 3 years ago

That was it! Thank you so much for getting back to me this is amazing!

DBraun commented 3 years ago

Great!