fdb/onnx-experiments

Experiments with the ONNX runtime for the web.

ONNX Experiments

This repository contains experiments on working with the ONNX format, and specifically using onnxruntime-web to run ONNX models in the browser.

Setup

I'm using Miniconda to manage my Python environment. I've created a new environment for this course:

conda create -y -n onnx-experiments python=3.11
conda activate onnx-experiments
conda install -y -c fastai -c conda-forge fastai jupyterlab onnx onnxruntime

To activate the environment:

conda activate onnx-experiments
jupyter lab

Training

I trained a simple image recognition model using the fast.ai appoach. Code is in the not-hotdog-training.ipynb notebook.

References

• Converting a fastai model to ONNX (Jupyter notebook)
• Deploy ONNX Runtime on the web
• Loading the images and converting them to tensors