74 lines
1.8 KiB
C++
74 lines
1.8 KiB
C++
#pragma once
|
|
#include <string>
|
|
#include <vector>
|
|
#include <argp.h>
|
|
#include <iostream>
|
|
#include <filesystem>
|
|
#include <opencv2/core/types.hpp>
|
|
#include "log.h"
|
|
|
|
const char *argp_program_version = "AIImagePreprocesses";
|
|
const char *argp_program_bug_address = "<carl@uvos.xyz>";
|
|
static char doc[] = "Application that trainsforms images into formats, sizes and aspect ratios required for ai training";
|
|
static char args_doc[] = "FILE(S)";
|
|
|
|
static struct argp_option options[] =
|
|
{
|
|
{"verbose", 'v', 0, 0, "Show debug messages" },
|
|
{"quiet", 'q', 0, 0, "only output data" },
|
|
{"model", 'm', "[FILENAME]", 0, "YoloV8 model to use for detection" },
|
|
{"classes", 'c', "[FILENAME]", 0, "classes text file to use" },
|
|
{"out", 'o', "[DIRECTORY]", 0, "directory whre images are to be saved" },
|
|
{"debug", 'd', 0, 0, "output debug images" },
|
|
{"seam-carving", 's', 0, 0, "model to train"},
|
|
{0}
|
|
};
|
|
|
|
struct Config
|
|
{
|
|
std::vector<std::filesystem::path> imagePaths;
|
|
std::filesystem::path modelPath;
|
|
std::filesystem::path classesPath;
|
|
std::filesystem::path outputDir;
|
|
bool seamCarving = false;
|
|
bool debug = false;
|
|
cv::Size targetSize = cv::Size(512, 512);
|
|
};
|
|
|
|
static error_t parse_opt (int key, char *arg, struct argp_state *state)
|
|
{
|
|
Config *config = reinterpret_cast<Config*>(state->input);
|
|
switch (key)
|
|
{
|
|
case 'q':
|
|
Log::level = Log::ERROR;
|
|
break;
|
|
case 'v':
|
|
Log::level = Log::DEBUG;
|
|
break;
|
|
case 'm':
|
|
config->modelPath = arg;
|
|
break;
|
|
case 'c':
|
|
config->classesPath = arg;
|
|
break;
|
|
case 'd':
|
|
config->debug = true;
|
|
break;
|
|
case 'o':
|
|
config->outputDir.assign(arg);
|
|
break;
|
|
case 's':
|
|
config->seamCarving = true;
|
|
break;
|
|
case ARGP_KEY_ARG:
|
|
config->imagePaths.push_back(arg);
|
|
break;
|
|
default:
|
|
return ARGP_ERR_UNKNOWN;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static struct argp argp = {options, parse_opt, args_doc, doc};
|