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
deactivateLa libreria compilata si trova in target/wheels.
| MNIST-12 | GoogLeNet-12 |