SuperResolutionTrainFunctionsDLL.h

Go to the documentation of this file.

 
#pragma once
 
#ifdef __linux__
    #define DLLTYPE_SUPER_RESOLUTION
#elif _WIN32
    #ifdef DLLEXPORT_SUPER_RESOLUTION
    #define DLLTYPE_SUPER_RESOLUTION extern "C" __declspec(dllexport)
    #else
    #define DLLTYPE_SUPER_RESOLUTION extern "C" __declspec(dllimport)
    #endif
#endif
 
#include "CommonHeaderDLL.h"
 
DLLTYPE_SUPER_RESOLUTION eErrorType
SR_SupResAlignImage(
    const wchar_t* sourceDir,
    const wchar_t* targetDir,
    const wchar_t* sourceResultDir,
    const wchar_t* targetResultDir,
    const float* roiParams
);
 
DLLTYPE_SUPER_RESOLUTION eErrorType
SR_ApplySupResTrain(
    const wchar_t* trainSourceDir, 
    const wchar_t* trainTargetDir,
    const float* roiParams,
    const wchar_t* initialModelFilePath,
    const wchar_t* saveModelDirPath, 
    const wchar_t* saveModelPrefixName,
    const int numIter=100000,
    const int checkpointInterval=10000,
    const int printInterval=1000,
    const int batchSize=4,
    const int cropSize=256,
    const int gpuIndex=0,
    const wchar_t* validationSourceDir=nullptr,
    const wchar_t* validationResultDir=nullptr
);