harisreedhar / Face-Upscalers-ONNX

ONNX-Powered Inference for State-of-the-Art Face Upscalers
81 stars 16 forks source link

size reduced models #1

Open ragavendranbala opened 1 year ago

ragavendranbala commented 1 year ago

I'm planning to create an app in Unity3D using barracuda. the model size seems larger than expected. Can you help me to create a low size models like tflite?

harisreedhar commented 1 year ago

These models are converted directly from pytorch without any optimization. Take a look at this repo which provides set of tools for optimizing onnx models.

Keep in mind that i don't own these models i just converted them that's it. So if you use it in production you should follow original model LICENSE.

couldnt-find-good-name commented 1 year ago

Hi, Can you tell how to convert .pth to .onnx. i tried googling but unable to decode from tutorials. Note, iam using win11.