tipe/test/cnn_utils.c

42 lines
1.3 KiB
C

#include <stdlib.h>
#include <stdio.h>
#include "../src/common/include/colors.h"
#include "../src/cnn/include/creation.h"
#include "../src/cnn/include/models.h"
#include "../src/cnn/include/utils.h"
#include "../src/cnn/include/free.h"
int main() {
printf("Création du réseau\n");
Network* network = create_network_lenet5(0, 0, 3, 2, 32, 1);
Network* network2 = create_network_lenet5(0, 0, 3, 2, 32, 1);
printf(GREEN "OK\n" RESET);
printf("Copie du réseau via copy_network\n");
Network* network_cp = copy_network(network);
printf(GREEN "OK\n" RESET);
printf("Vérification de l'égalité des réseaux\n");
if (! equals_networks(network, network_cp)) {
printf_error(RED "Les deux réseaux obtenus ne sont pas égaux.\n" RESET);
exit(1);
}
printf(GREEN "OK\n" RESET);
printf("Copie du réseau via copy_network_parameters\n");
copy_network_parameters(network, network2);
printf(GREEN "OK\n" RESET);
printf("Vérification de l'égalité des réseaux\n");
if (! equals_networks(network, network2)) {
printf_error(RED "Les deux réseaux obtenus ne sont pas égaux.\n" RESET);
exit(1);
}
printf(GREEN "OK\n" RESET);
free_network(network_cp);
free_network(network2);
free_network(network);
return 0;
}