diff --git a/benchmarks/tract-onnx-image-classification/benchmark.stdout.expected b/benchmarks/tract-onnx-image-classification/benchmark.stdout.expected index d319a28..49ac84d 100644 --- a/benchmarks/tract-onnx-image-classification/benchmark.stdout.expected +++ b/benchmarks/tract-onnx-image-classification/benchmark.stdout.expected @@ -1 +1 @@ -[Classification { label: "tiger", score: 17.559244 }, Classification { label: "tiger cat", score: 14.740076 }, Classification { label: "zebra", score: 12.357242 }] +[Classification { label: "tiger", score: 17.56 }, Classification { label: "tiger cat", score: 14.74 }, Classification { label: "zebra", score: 12.36 }] diff --git a/benchmarks/tract-onnx-image-classification/benchmark.wasm b/benchmarks/tract-onnx-image-classification/benchmark.wasm index e236ebd..c10a6c7 100755 Binary files a/benchmarks/tract-onnx-image-classification/benchmark.wasm and b/benchmarks/tract-onnx-image-classification/benchmark.wasm differ diff --git a/benchmarks/tract-onnx-image-classification/rust-benchmark/src/lib.rs b/benchmarks/tract-onnx-image-classification/rust-benchmark/src/lib.rs index 3d41ed9..7b4ecf0 100644 --- a/benchmarks/tract-onnx-image-classification/rust-benchmark/src/lib.rs +++ b/benchmarks/tract-onnx-image-classification/rust-benchmark/src/lib.rs @@ -78,7 +78,7 @@ fn classify(image: &[u8]) -> Result, anyhow::Error> { .take(3) .map(|(score, i)| Classification { label: LABELS[*i as usize].to_string(), - score: **score, + score: format!("{:.*}", 2, **score).parse().unwrap(), }) .collect(); Ok(labels)