Progetto 2.1 - Configurazione ed esecuzione ottimizzata di una rete neurale generica descritta con standard ONNX
# Build
cargo build --release
# Run
./target/release/onnx-cli <model> --input <input_file> [--output <output_file>]
model
: Modello in formato .onnx da leggereinput_file
: Dati in ingresso al modello in formato .pboutput_file
: Eventuali dati di output attesi in formato .pb, stampati insieme all'output effettivo del modello, utile per valutare la correttezza dell'inferenza
Nel file tests/models.rs
sono presenti i test relativi ai 2 modelli usati per lo sviluppo dell'applicazione:
mnist
: Test relativo al modello MNIST-12googlenet
: Test relativo al modello GoogLeNet-12
Per valutare la correttezza dei modelli si assume che il risultato sia uguale valutandolo fino alla terza cifra significativa
# Build
python/build.sh
# Start virtual environment
source .env/bin/activate
# Run python program
python python/test.py
# Close virtual environment when you want to exit
deactivate
La libreria compilata si trova in target/wheels
.
MNIST-12 | GoogLeNet-12 |