56 lines
1.1 KiB
C++
56 lines
1.1 KiB
C++
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <unistd.h>
|
|
|
|
#include "uNet.h"
|
|
#include "util.h"
|
|
|
|
// Global variables
|
|
int N = 1;
|
|
int random_seed = 1;
|
|
int print_max = 8;
|
|
int MAX_LEN = 10;
|
|
char *parameter_fname;
|
|
char *output_fname;
|
|
|
|
|
|
int main(int argc, char **argv) {
|
|
|
|
check_and_parse_args(argc, argv);
|
|
|
|
// Initialize model
|
|
styler_initialize(N, random_seed, parameter_fname);
|
|
|
|
float *random_floats = nullptr;
|
|
char *output = nullptr;
|
|
|
|
// Initialize input and output
|
|
random_floats = (float *)malloc(N * MAX_LEN * sizeof(float));
|
|
output = (char *)malloc(N * (MAX_LEN + 1) * sizeof(char));
|
|
srand(random_seed);
|
|
for (int i = 0; i < N * MAX_LEN; i++) {
|
|
random_floats[i] = ((float)rand()) / ((float)RAND_MAX);
|
|
}
|
|
|
|
printf("Styling %d images...", N);
|
|
fflush(stdout);
|
|
|
|
// Styling images and measure time
|
|
double styler_st = get_time();
|
|
|
|
styler(N, random_floats, output);
|
|
|
|
double styler_en = get_time();
|
|
|
|
double elapsed_time = styler_en - styler_st;
|
|
printf("Done!\n");
|
|
|
|
// Print first few result
|
|
print_first_few_result(output, print_max, elapsed_time);
|
|
|
|
// Finalize program
|
|
styler_finalize();
|
|
}
|
|
|
|
|