296 - Converting keras trained model to ONNX format - Image Classification example

Описание

Code generated in the video can be downloaded from here:
https://github.com/bnsreenu/python_for_microscopists/blob/master/296-Converting%20keras-trained%20model%20to%20ONNX%20format-Img%20Classification.py

All other code:
https://github.com/bnsreenu/python_for_microscopists

This tutorial covers the following topics...
1. Training a keras model for multiclass classification using the cifar10 dataset.
2. Saving the keras model as .h5
3. Classification using the saved keras .h5 model
4. Converting the keras model to onnx format
5.Classification using the onnx model (and comparison to the keras results)

pip install keras2onnx #For older tensorflow (up to 2.3.1)
pip install tf2onnx #For newer tensorflow (I tested on 2.4.4)
pip install onnxruntime
pip install h5py