mrdbourke / nutrify

Take a photo of food and learn about it.
https://nutrify.app
MIT License
178 stars 33 forks source link

Create `export.py` (or something similar) for exporting models from PyTorch -> CoreML #62

Open mrdbourke opened 1 year ago

mrdbourke commented 1 year ago

Could have multiple export types as well:

More:

mrdbourke commented 1 year ago

See this notebook for an example conversion: https://github.com/mrdbourke/nutrify/blob/main/foodvision/notebooks/06_export_model.ipynb

Note: Prediction/evaluation can only take place on macOS with CoreML, should setup some code to make a prediction over ~100 images with CoreML & PyTorch for a sanity check on how the model is performing.