Downloads · 30 days
0
Cat_Dog_Breeds.ONNX
ScottMueller/Cat_Dog_Breeds.ONNX
Cat_Dog_Breeds.ONNX is a machine learning model from ScottMueller. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
A simple single label classification model, ResNet18, to predict the cat or dog breed from the provided image. The model was created in Fast.ai and exported to ONNX using PyTorch's ONNX export capabilities.
Access
Public
Updated Aug 8, 2022
Repo size
47.2 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.onnx47.2 MB · 100%
At a glance
- License
- mit
- Access
- Public
- Created
- Aug 8, 2022
- Updated
- Aug 8, 2022
- SHA
- b0518d13
Model card
From the Hugging Face model README
A simple single label classification model, ResNet18, to predict the cat or dog breed from the provided image. The model was created in Fast.ai and exported to ONNX using PyTorch's ONNX export capabilities.
The source dataset is the OXFORD-IIIT PET. Omkar M Parkhi, Andrea Vedaldi, Andrew Zisserman and C. V. Jawahar We have created a 37 category pet dataset with roughly 200 images for each class. The images have a large variations in scale, pose and lighting. All images havean associated ground truth annotation of breed, head ROI, and pixel level trimap segmentation.
The ONNX model can be used in other frameworks like Elixir's Axon. An example of converting the ONNX model into Axon can be found at: https://github.com/elixir-nx/axon/tree/main/notebooks/onnx_to_axon.livemd.
- License
- mit
- Created
- Aug 8, 2022
- Updated
- Aug 8, 2022