Skip to content

Wrapping an ONNX Deep Learning model into a standalone Rust executable

Notifications You must be signed in to change notification settings

sonovice/tract-testing

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Test case for the tract library

This is an experiment on how to wrap a PyTorch-trained ONNX U-Net model into a standalone Rust executable.

Notes

  • Input/Output image size is fixed for now (1211 x 900), but arbitrary sizes can be extended easily.
  • Execution is entirely single core! That means rather high latency, but almost linear scalability.
  • The model could be embedded in the executable as well using the include_bytes! macro.
  • For the curious: The bundled mini model does some (bad) pixel segmentation on binary sheet music images. The full model is alot bigger and better.

About

Wrapping an ONNX Deep Learning model into a standalone Rust executable

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages