Skip to content

Latest commit

 

History

History

pytorch-resnet

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Example of exporting a PyTorch to ONNX format, then performing inference with tract.

Export Model:

python export.py

Inference on elephants.jpg:

cargo run
result: Some((22.08386, 102))

Predicts class 102 (tusker).