qualitybrisque.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_QUALITY_QUALITYBRISQUE_HPP




#define OPENCV_QUALITY_QUALITYBRISQUE_HPP








#include "qualitybase.hpp"




#include "opencv2/ml.hpp"








namespace

cv



{



namespace
quality


{







class
CV_EXPORTS_W
QualityBRISQUE
:
public
QualityBase
{



public:







CV_WRAP
cv::Scalar
compute( InputArray img ) CV_OVERRIDE;







CV_WRAP
static
Ptr<QualityBRISQUE>
create(
const
cv::String& model_file_path,
const
cv::String& range_file_path );







CV_WRAP
static
Ptr<QualityBRISQUE>
create(
const
cv::Ptr<cv::ml::SVM>& model,
const
cv::Mat& range );







CV_WRAP
static
cv::Scalar
compute( InputArray img,
const
cv::String& model_file_path,
const
cv::String& range_file_path );







CV_WRAP
static
void
computeFeatures(InputArray img,
OutputArray
features);







protected:







cv::Ptr<cv::ml::SVM>
_model =
nullptr;



cv::Mat
_range;







QualityBRISQUE(
const
cv::String& model_file_path,
const
cv::String& range_file_path );







QualityBRISQUE(const
cv::Ptr<cv::ml::SVM>& model,
const
cv::Mat& range )



: _model{ model }



, _range{ range }



{}






};
// QualityBRISQUE



}
// quality



}
// cv




#endif