97 lines
2.7 KiB
C++
97 lines
2.7 KiB
C++
#include "util.h"
|
|
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
|
|
// Defined in main.cpp
|
|
extern int N;
|
|
extern char *parameter_fname;
|
|
extern char *output_fname;
|
|
extern char *input_fname;
|
|
|
|
void read_binary(void *dst, const char *filename, size_t *size) {
|
|
size_t size_;
|
|
FILE *f = fopen(filename, "rb");
|
|
CHECK_ERROR(f != NULL, "Failed to read %s", filename);
|
|
fseek(f, 0, SEEK_END);
|
|
size_ = ftell(f);
|
|
rewind(f);
|
|
size_t ret = fread(dst, 1, size_, f);
|
|
fclose(f);
|
|
CHECK_ERROR(size_ == ret, "Failed to read %ld bytes from %s", size_,
|
|
filename);
|
|
if (size != NULL) *size = (size_t)(size_ / 4); // float
|
|
}
|
|
|
|
void *read_binary(const char *filename, size_t *size) {
|
|
size_t size_;
|
|
FILE *f = fopen(filename, "rb");
|
|
CHECK_ERROR(f != NULL, "Failed to read %s", filename);
|
|
fseek(f, 0, SEEK_END);
|
|
size_ = ftell(f);
|
|
rewind(f);
|
|
void *buf = malloc(size_);
|
|
size_t ret = fread(buf, 1, size_, f);
|
|
fclose(f);
|
|
CHECK_ERROR(size_ == ret, "Failed to read %ld bytes from %s", size_,
|
|
filename);
|
|
if (size != NULL) *size = (size_t)(size_ / 4); // float
|
|
return buf;
|
|
}
|
|
|
|
void write_binary(void *dst, const char *filename, size_t size) {
|
|
FILE *output_fp = (FILE *) fopen(output_fname, "wb");
|
|
fwrite(dst, sizeof(float), size, output_fp);
|
|
fclose(output_fp);
|
|
}
|
|
|
|
double get_time() {
|
|
struct timespec tv;
|
|
clock_gettime(CLOCK_MONOTONIC, &tv);
|
|
return tv.tv_sec + tv.tv_nsec * 1e-9;
|
|
}
|
|
|
|
void print_usage_exit(int argc, char **argv) {
|
|
printf("Usage %s [parameter bin] [input bin] [output] [N]\n", argv[0]);
|
|
printf(" parameter bin: File containing DNN parameters\n");
|
|
printf(" input bin : File containing input images\n");
|
|
printf(" output: File to write results\n");
|
|
printf(" N: Number of images to mask\n");
|
|
EXIT(0);
|
|
}
|
|
|
|
void check_and_parse_args(int argc, char **argv) {
|
|
if (argc != 5) print_usage_exit(argc, argv);
|
|
|
|
int c;
|
|
while ((c = getopt(argc, argv, "h")) != -1) {
|
|
switch (c) {
|
|
case 'h': break;
|
|
default: print_usage_exit(argc, argv);
|
|
}
|
|
}
|
|
|
|
parameter_fname = argv[1];
|
|
input_fname = argv[2];
|
|
output_fname = argv[3];
|
|
N = atoi(argv[4]);
|
|
}
|
|
|
|
void print_model() {
|
|
printf("\n Model : U-Net\n");
|
|
printf(
|
|
"------------------------------------------------------------------\n");
|
|
printf(" Automatically identify the boundaries of the images in input.bin\n");
|
|
printf(
|
|
"==================================================================\n");
|
|
printf(" Number of input images : %d\n", N);
|
|
printf(" Parameter file : %s\n", parameter_fname);
|
|
printf(" Input file : %s\n", input_fname);
|
|
printf(" Output file to write results : %s\n", output_fname);
|
|
printf(
|
|
"==================================================================\n");
|
|
}
|