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

Add export.c

Browse Source
This commit is contained in:
augustin64 2023-02-05 16:21:41 +01:00
parent b18fd7aab9
commit 635b46bdf3
2 changed files with 102 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
Makefile
View File

@ -30,8 +30,8 @@ LD_CFLAGS = -lm -lpthread -ljpeg -fopenmp
LD_NVCCFLAGS = -ljpeg -Xcompiler -fopenmp
# Compilation flag
CFLAGS = -Wall -Wextra -std=gnu99
NVCCFLAGS =
CFLAGS = -Wall -Wextra -std=gnu99 -g
NVCCFLAGS = -g
# Remove warnings about unused variables, functions, ...
# -Wno-unused-parameter -Wno-unused-function -Wno-unused-variable -Wno-unused-but-set-variable
# Compile with debug
@ -69,7 +69,7 @@ $(BUILDDIR)/mnist_%.o: $(MNIST_SRCDIR)/%.c $(MNIST_SRCDIR)/include/%.h
#
# Build cnn
#
cnn: $(BUILDDIR)/cnn-main $(BUILDDIR)/cnn-main-cuda $(BUILDDIR)/cnn-preview;
cnn: $(BUILDDIR)/cnn-main $(BUILDDIR)/cnn-main-cuda $(BUILDDIR)/cnn-preview $(BUILDDIR)/cnn-export;
$(BUILDDIR)/cnn-main: $(CNN_SRCDIR)/main.c \
$(BUILDDIR)/cnn_train.o \
@ -119,6 +119,9 @@ endif
$(BUILDDIR)/cnn-preview: $(CNN_SRCDIR)/preview.c $(BUILDDIR)/cnn_jpeg.o $(BUILDDIR)/colors.o $(BUILDDIR)/utils.o
$(CC) $^ -o $@ $(CFLAGS) $(LD_CFLAGS)
$(BUILDDIR)/cnn-export: $(CNN_SRCDIR)/export.c $(BUILDDIR)/cnn_free.o $(BUILDDIR)/cnn_neuron_io.o $(BUILDDIR)/utils.o
$(CC) $^ -o $@ $(CFLAGS) $(LD_CFLAGS)
$(BUILDDIR)/cnn_%.o: $(CNN_SRCDIR)/%.c $(CNN_SRCDIR)/include/%.h
$(CC) -c $< -o $@ $(CFLAGS)

96
src/cnn/export.c Normal file
View File

@ -0,0 +1,96 @@
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include "include/free.h"
#include "include/struct.h"
#include "include/neuron_io.h"
void help(char* call) {
printf("Usage: %s ( print-poids-kernel-cnn ) [OPTIONS]\n\n", call);
}
void print_poids_ker_cnn(char* modele) {
Network* network = read_network(modele);
int vus = 0;
printf("{\n");
for (int i=0; i < network->max_size-1; i++) {
Kernel_cnn* kernel_cnn = network->kernel[i]->cnn;
if (!(!kernel_cnn)) {
if (vus != 0) {
printf(",");
}
vus++;
printf("\t\"%d\":[\n", i);
for (int i=0; i < kernel_cnn->rows; i++) {
printf("\t\t[\n");
for (int j=0; j < kernel_cnn->columns; j++) {
printf("\t\t\t[\n");
for (int k=0; k < kernel_cnn->k_size; k++) {
printf("\t\t\t\t[");
for (int l=0; l < kernel_cnn->k_size; l++) {
printf("%lf", kernel_cnn->w[i][j][k][l]);
if (l != kernel_cnn->k_size-1) {
printf(", ");
}
}
printf(" ]");
if (k != kernel_cnn->k_size-1) {
printf(",");
}
printf("\n");
}
printf("\t\t\t]");
if (j != kernel_cnn->columns-1) {
printf(",");
}
printf("\n");
}
printf("\t\t]");
if (i != kernel_cnn->rows-1) {
printf(",");
}
printf("\n");
}
printf("\t]\n");
}
}
printf("}\n");
free_network(network);
}
int main(int argc, char* argv[]) {
if (argc < 2) {
printf("Pas d'action spécifiée\n");
help(argv[0]);
return 1;
}
if (! strcmp(argv[1], "print-poids-kernel-cnn")) {
char* modele = NULL; // Fichier contenant le modèle
int i = 2;
while (i < argc) {
if ((! strcmp(argv[i], "--modele"))||(! strcmp(argv[i], "-m"))) {
modele = argv[i+1];
i += 2;
} else {
printf("Option choisie inconnue: %s\n", argv[i]);
i++;
}
}
if (!modele) {
printf("Pas de modèle à utiliser spécifié.\n");
return 1;
}
print_poids_ker_cnn(modele);
return 0;
}
printf("Option choisie non reconnue: %s\n", argv[1]);
help(argv[0]);
return 1;
}