Skip to content

Added ONNX export capability for crepe models #105

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

joeloftusdev
Copy link

@joeloftusdev joeloftusdev commented Apr 4, 2025

Add ONNX export capability for CREPE models

This commit adds support for exporting CREPE models to ONNX format:

  1. Add export_model_to_onnx() function to core.py that converts Keras models to ONNX
  2. Create command-line script (onnx_export.py) to easily export models with different capacities
  3. Update .gitignore to exclude generated ONNX model files

The ONNX export enables using CREPE models in non-TensorFlow environments and
optimizing with tools like ONNX Runtime for deployment across various platforms.

Usage examples:

python -m crepe.onnx_export tiny  # Creates model-tiny.onnx
python -m crepe.onnx_export full -o custom_name.onnx

@joeloftusdev
Copy link
Author

example usage here: https://github.com/joeloftusdev/crepe.cpp

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant