[Request] ONNX version

#1
by jspsoli - opened

Hi! Thank you for working on this model. It seems far better at furries and ferals than ZED's. A bit worse on humans though.
Could you please consider making a .onnx version release? Can't run it on my AMD gpu in pytorch format.

Added!

Thanks alot! I'll try it right away.

Edit: Got it working with directml GPU and for me its a literal 10x speed boost over CPU. Exact same output tags as the pytorch version for the same thresholds - flawless conversion :D Thanks again!

jspsoli changed discussion status to closed

Sign up or log in to comment