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

Add print-images

Browse Source
This commit is contained in:
augustin64 2022-07-01 16:16:55 +02:00
parent 9a032a4933
commit 9df46ff801
1 changed files with 59 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
src/mnist/utils.c
View File

@ -25,6 +25,8 @@ void help(char* call) {
printf("\tpatch-network:\n");
printf("\t\t--network | -n [FILENAME]\tFichier contenant le réseau de neurones.\n");
printf("\t\t--delta | -d [FILENAME]\tFichier de patch à utiliser.\n");
printf("\tprint-images:\n");
printf("\t\t--images | -i [FILENAME]\tFichier contenant les images.\n");
}
@ -132,6 +134,44 @@ void patch_stored_network(char* network_filename, char* delta_filename) {
}
void print_images(char* filename) {
int* parameters = read_mnist_images_parameters(filename);
int nb_elem = parameters[0];
int width = parameters[1];
int height = parameters[2];
free(parameters);
int*** images = read_mnist_images(filename);
printf("[\n");
for (int i=0; i < nb_elem; i++) {
printf("\t[\n");
for (int j=0; j < height; j++) {
printf("\t\t[");
for (int k=0; k < width; k++) {
if (k != width -1)
printf("%d, ", images[i][j][k]);
else
printf("%d", images[i][j][k]);
}
if (j != height -1)
printf("],\n");
else
printf("]\n");
free(images[i][j]);
}
if (i != nb_elem -1)
printf("\t],\n");
else
printf("\t]\n");
free(images[i]);
}
free(images);
printf("]\n");
}
int main(int argc, char* argv[]) {
if (argc < 2) {
printf("Pas d'action spécifiée\n");
@ -236,6 +276,24 @@ int main(int argc, char* argv[]) {
}
patch_stored_network(network, delta);
exit(0);
} else if (! strcmp(argv[1], "print-images")) {
char* images = NULL;
int i = 2;
while (i < argc) {
if ((! strcmp(argv[i], "--images"))||(! strcmp(argv[i], "-i"))) {
images = argv[i+1];
i += 2;
} else {
printf("%s : Argument non reconnu\n", argv[i]);
i++;
}
}
if (!images) {
printf("--images: Argument obligatoire.\n");
exit(1);
}
print_images(images);
exit(0);
}
printf("Option choisie non reconnue: %s\n", argv[1]);
help(argv[0]);