augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-01-23 23:26:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add recognize

Browse Source
This commit is contained in:
augustin64 2022-04-08 16:07:44 +02:00
parent c3d07ad638
commit 5bb43ff5a7
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/mnist/main.c
View File

@ -35,6 +35,7 @@ void ecrire_image_dans_reseau(int** image, Reseau* reseau, int height, int width
void train(int batches, int couches, int neurons, char* recovery, char* image_file, char* label_file, char* out) {
// Entraînement du réseau sur le set de données MNIST
Reseau* reseau;
//int* repartition = malloc(sizeof(int)*couches);
@ -73,10 +74,32 @@ void train(int batches, int couches, int neurons, char* recovery, char* image_fi
backward_propagation(reseau, sortie_voulue);
}
printf("\n");
modification_du_reseau_neuronal(reseau);
ecrire_reseau(out, reseau);
}
}
void recognize(char* modele, char* entree, char* sortie) {
Reseau* reseau = lire_reseau(modele);
int* parameters = read_mnist_images_parameters(entree);
int nb_images = parameters[0];
int height = parameters[1];
int width = parameters[2];
int*** images = read_mnist_images(entree);
for (int i=0; i < nb_images; i++) {
printf("Image %d\n", i);
ecrire_image_dans_reseau(images[i], reseau, height, width);
forward_propagation(reseau);
for (int j=0; j < reseau->couches[reseau->nb_couches-1]->nb_neurones; j++) {
printf("Probabilité %d: %f\n", j, reseau->couches[reseau->nb_couches-1]->neurones[j]->activation);
}
}
}
int main(int argc, char* argv[]) {
if (argc < 2) {
@ -169,6 +192,7 @@ int main(int argc, char* argv[]) {
if (! out) {
out = "text";
}
recognize(modele, in, out);
// Reconnaissance puis affichage des données sous le format spécifié
exit(0);
}