toy project
resnet
cifar10
classification
pytorch
torchvision
onxx
gradio
Use cifar10 dataset bird and airplane to train a model, then convert the model to onnx format, and deploy the model with gradio.
conda create -n flyorbird python=3.10
pip install torch torchvision gradio numpy onnxruntime-gpu
Note: maybe is not enough, place by yourself.
python train.py
pretrained model weights can get for release
python test.py
python convert2onnx.py
onxx model can get for release
python deploy.py