mirror of
https://github.com/lltcggie/waifu2x-caffe.git
synced 2025-06-26 05:32:47 +00:00
33 lines
1.2 KiB
C
33 lines
1.2 KiB
C
|
#pragma once
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
#include <vector>
|
||
|
|
#include <utility>
|
||
|
|
#include <functional>
|
||
|
|
|
||
|
|
enum eWaifu2xError
|
||
|
|
{
|
||
|
|
eWaifu2xError_OK = 0,
|
||
|
|
eWaifu2xError_Cancel,
|
||
|
|
eWaifu2xError_InvalidParameter,
|
||
|
|
eWaifu2xError_FailedOpenInputFile,
|
||
|
|
eWaifu2xError_FailedOpenOutputFile,
|
||
|
|
eWaifu2xError_FailedOpenModelFile,
|
||
|
|
eWaifu2xError_FailedParseModelFile,
|
||
|
|
eWaifu2xError_FailedConstructModel,
|
||
|
|
eWaifu2xError_FailedProcessCaffe,
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef std::pair<std::string, std::string> InputOutputPathPair;
|
||
|
|
typedef std::pair<InputOutputPathPair, eWaifu2xError> PathAndErrorPair;
|
||
|
|
typedef std::function<bool()> waifu2xCancelFunc;
|
||
|
|
typedef std::function<void(const int ProgressFileMax, const int ProgressFileNow)> waifu2xProgressFunc;
|
||
|
|
|
||
|
|
bool can_use_cuDNN();
|
||
|
|
|
||
|
|
// mode: noise or scale or noise_scale or auto_scale
|
||
|
|
// process: cpu or gpu or cudnn
|
||
|
|
eWaifu2xError waifu2x(int argc, char** argv,
|
||
|
|
const std::vector<InputOutputPathPair> &file_paths, const std::string &mode, const int noise_level, const double scale_ratio, const std::string &model_dir, const std::string &process,
|
||
|
|
std::vector<PathAndErrorPair> &errors, const waifu2xCancelFunc cancel_func = nullptr, const waifu2xProgressFunc progress_func = nullptr);
|