OpenCV  4.2.0
Open Source Computer Vision
Public Member Functions | List of all members
cv::dnn_superres::DnnSuperResImpl Class Reference

A class to upscale images via convolutional neural networks. The following four models are implemented: More...

#include <opencv2/dnn_superres.hpp>

Public Member Functions

 DnnSuperResImpl ()
 Empty constructor. More...
 
 DnnSuperResImpl (const std::string &algo, int scale)
 Constructor which immediately sets the desired model. More...
 
std::string getAlgorithm ()
 Returns the scale factor of the model: More...
 
int getScale ()
 Returns the scale factor of the model: More...
 
void readModel (const std::string &path)
 Read the model from the given path. More...
 
void readModel (const std::string &weights, const std::string &definition)
 Read the model from the given path. More...
 
void setModel (const std::string &algo, int scale)
 Set desired model. More...
 
void upsample (InputArray img, OutputArray result)
 Upsample via neural network. More...
 
void upsampleMultioutput (InputArray img, std::vector< Mat > &imgs_new, const std::vector< int > &scale_factors, const std::vector< String > &node_names)
 Upsample via neural network of multiple outputs. More...
 

Detailed Description

A class to upscale images via convolutional neural networks. The following four models are implemented:

Constructor & Destructor Documentation

◆ DnnSuperResImpl() [1/2]

cv::dnn_superres::DnnSuperResImpl::DnnSuperResImpl ( )

Empty constructor.

◆ DnnSuperResImpl() [2/2]

cv::dnn_superres::DnnSuperResImpl::DnnSuperResImpl ( const std::string &  algo,
int  scale 
)

Constructor which immediately sets the desired model.

Parameters
algoString containing one of the desired models:
  • edsr
  • espcn
  • fsrcnn
  • lapsrn
scaleInteger specifying the upscale factor

Member Function Documentation

◆ getAlgorithm()

std::string cv::dnn_superres::DnnSuperResImpl::getAlgorithm ( )

Returns the scale factor of the model:

Returns
Current algorithm.

◆ getScale()

int cv::dnn_superres::DnnSuperResImpl::getScale ( )

Returns the scale factor of the model:

Returns
Current scale factor.

◆ readModel() [1/2]

void cv::dnn_superres::DnnSuperResImpl::readModel ( const std::string &  path)

Read the model from the given path.

Parameters
pathPath to the model file.

◆ readModel() [2/2]

void cv::dnn_superres::DnnSuperResImpl::readModel ( const std::string &  weights,
const std::string &  definition 
)

Read the model from the given path.

Parameters
weightsPath to the model weights file.
definitionPath to the model definition file.

◆ setModel()

void cv::dnn_superres::DnnSuperResImpl::setModel ( const std::string &  algo,
int  scale 
)

Set desired model.

Parameters
algoString containing one of the desired models:
  • edsr
  • espcn
  • fsrcnn
  • lapsrn
scaleInteger specifying the upscale factor

◆ upsample()

void cv::dnn_superres::DnnSuperResImpl::upsample ( InputArray  img,
OutputArray  result 
)

Upsample via neural network.

Parameters
imgImage to upscale
resultDestination upscaled image

◆ upsampleMultioutput()

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

Upsample via neural network of multiple outputs.

Parameters
imgImage to upscale
imgs_newDestination upscaled images
scale_factorsScaling factors of the output nodes
node_namesNames of the output nodes in the neural network

The documentation for this class was generated from the following file: