ScottMueller commited on
Commit
87add5c
1 Parent(s): 70a46ab

Update the Readme file

Browse files

Added references to how the ONNX model file was created. Added reference to the datasource used in training. Also, have a link to how the ONNX file can be converted into an Axon model.

Files changed (1) hide show
  1. README.md +10 -0
README.md CHANGED
@@ -1,3 +1,13 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
  ---
4
+ A simple single label classification model, ResNet18, to predict whether the provided image is a cat or a dog. The model was created in Fast.ai
5
+ and exported to ONNX using PyTorch's ONNX export capabilities.
6
+
7
+ The source dataset is the OXFORD-IIIT PET. Omkar M Parkhi, Andrea Vedaldi, Andrew Zisserman and C. V. Jawahar
8
+ We have created a 37 category pet dataset with roughly 200 images for each class.
9
+ The images have a large variations in scale, pose and lighting. All images havean
10
+ associated ground truth annotation of breed, head ROI, and pixel level trimap segmentation.
11
+
12
+ 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:
13
+ https://github.com/elixir-nx/axon/tree/main/notebooks/onnx_to_axon.livemd.