From fd7fe39e14a598b473695a9747b5e77b02a16f84 Mon Sep 17 00:00:00 2001 From: augustin64 Date: Tue, 26 Apr 2022 17:09:12 +0200 Subject: [PATCH] Update main.c --- src/mnist/main.c | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/mnist/main.c b/src/mnist/main.c index 761894b..0206cba 100644 --- a/src/mnist/main.c +++ b/src/mnist/main.c @@ -211,6 +211,8 @@ void test(char* modele, char* fichier_images, char* fichier_labels, bool preview int* parameters = read_mnist_images_parameters(fichier_images); int nb_images = parameters[0]; + int width = parameters[1]; + int height = parameters[2]; int*** images = read_mnist_images(fichier_images); float** resultats = recognize(modele, fichier_images); @@ -222,7 +224,7 @@ void test(char* modele, char* fichier_images, char* fichier_labels, bool preview accuracy += 1. / (float)nb_images; } else { printf("--- Image %d, %d --- Prévision: %d ---\n", i, labels[i], indice_max(resultats[i], nb_der_layer)); - print_image(28, 28, images[i], resultats[i]); + print_image(width, height, images[i], resultats[i]); } } printf("%d Images\tAccuracy: %0.1f%%\n", nb_images, accuracy*100);