dnn_superres.hpp

// This file is part of OpenCV project.




// It is subject to the license terms in the LICENSE file found in the top-level directory




// of this distribution and at http://opencv.org/license.html.








#ifndef _OPENCV_DNN_SUPERRES_HPP_




#define _OPENCV_DNN_SUPERRES_HPP_








#include "opencv2/core.hpp"




#include "opencv2/dnn.hpp"








namespace

cv



{



namespace
dnn_superres


{











class
CV_EXPORTS_W
DnnSuperResImpl



{



private:







dnn::Net
net;







std::string alg;
//algorithm








int
sc;
//scale factor








void
reconstruct_YCrCb(InputArray inpImg, InputArray origImg,
OutputArray
outpImg,
int
scale);







void
preprocess_YCrCb(InputArray inpImg,
OutputArray
outpImg);







public:







CV_WRAP
static
Ptr<DnnSuperResImpl>
create();







// /** @brief Empty constructor




//  */




DnnSuperResImpl();







DnnSuperResImpl(const
String& algo,
int
scale);







CV_WRAP
void
readModel(const
String& path);







void
readModel(const
String& weights,
const
String& definition);







CV_WRAP
void
setModel(const
String& algo,
int
scale);







CV_WRAP
void
setPreferableBackend(int
backendId);







CV_WRAP
void
setPreferableTarget(int
targetId);







CV_WRAP
void
upsample(InputArray img,
OutputArray
result);







CV_WRAP
void
upsampleMultioutput(InputArray img, std::vector<Mat> &imgs_new,
const
std::vector<int>& scale_factors,
const
std::vector<String>& node_names);







CV_WRAP
int
getScale();







CV_WRAP String getAlgorithm();


};










}}
// cv::dnn_superres::




#endif