search

CNChTu / FCPE

MIT License
83 stars 16 forks source link

onnx model required #9

Open ykk648 opened 6 months ago

CNChTu commented 5 months ago

Coming Soon. In fact, we have completed the code and will upload it after testing.

joe0327 commented 5 months ago

Hi，How's the progress?

deiteris commented 3 weeks ago

Since there isn't much progress, I've made an ONNX model of fcpe by decoupling mel extractor and fcpe model with this script: https://github.com/deiteris/voice-changer/blob/master-custom/server/utils/fcpe_onnx.py

It should be also possible to produce a combined model by returning float tensor instead of complex in torch.stft. https://github.com/CNChTu/FCPE/blob/main/torchfcpe/mel_extractor.py#L136-L139

There's a good example of how it's done for RMVPE algorithm: https://github.com/yxlllc/RMVPE/blob/main/export.py#L44-L57