32#include "digikam_opencv.h"
48 QList<QString> loadDetectionClasses();
50 std::vector<cv::Mat> preprocess(
const cv::Mat& inputImage);
51 std::vector<cv::Mat> preprocess(
const std::vector<cv::Mat>& inputBatchImages);
53 QList<QHash<QString, QVector<QRect> > > postprocess(
const std::vector<cv::Mat>& inputBatchImages,
54 const std::vector<cv::Mat>& outs)
const;
56 virtual QHash<QString, QVector<QRect> >
postprocess(
const cv::Mat& inputImage,
57 const cv::Mat& out)
const = 0;
59 std::vector<cv::String> getOutputsNames()
const;
65 virtual QHash<QString, QVector<QRect> >
detectObjects(
const cv::Mat& inputImage);
70 virtual QList<QHash<QString, QVector<QRect> > > detectObjects(
const std::vector<cv::Mat>& inputBatchImages);
75 virtual QList<QString> getPredefinedClasses()
const;
81 QList<QString> generateObjects(
const cv::Mat& inputImage);
87 QList<QList<QString> > generateObjects(
const std::vector<cv::Mat>& inputImage);
92 cv::Size getinputImageSize()
const;
96 double showInferenceTime();
108 float scaleFactor = 1.0F;
Definition dnnbasedetectormodel.h:40
static float scoreThreshold
Threshold for class detection score.
Definition dnnbasedetectormodel.h:102
cv::Size inputImageSize
Definition dnnbasedetectormodel.h:110
static int uiConfidenceThreshold
Threshold for bbox detection. It can be init and changed in the GUI.
Definition dnnbasedetectormodel.h:100
virtual ~DNNBaseDetectorModel()=default
virtual QHash< QString, QVector< QRect > > detectObjects(const cv::Mat &inputImage)
virtual bool loadModels()=0
cv::Scalar meanValToSubtract
Definition dnnbasedetectormodel.h:109
virtual QHash< QString, QVector< QRect > > postprocess(const cv::Mat &inputImage, const cv::Mat &out) const =0
static float nmsThreshold
Threshold for nms suppression.
Definition dnnbasedetectormodel.h:101
QList< QString > predefinedClasses
Definition dnnbasedetectormodel.h:111
Definition dnnmodelbase.h:35
Definition datefolderview.cpp:34