digiKam
Loading...
Searching...
No Matches
Digikam::DNNSFaceExtractor Class Reference

#include <dnnsfaceextractor.h>

+ Inheritance diagram for Digikam::DNNSFaceExtractor:

Public Member Functions

virtual cv::Mat alignFace (const cv::Mat &inputImage) const override
 
 DNNSFaceExtractor ()
 
virtual cv::Mat getFaceEmbedding (const cv::Mat &faceImage) override
 
bool loadModels () override
 
 ~DNNSFaceExtractor () override
 

Static Public Member Functions

static double cosineDistance (const std::vector< float > &v1, const std::vector< float > &v2)
 
static std::vector< float > decodeVector (const QJsonArray &json)
 
static QJsonArray encodeVector (const std::vector< float > &vector)
 
static double L2squareDistance (const std::vector< float > &v1, const std::vector< float > &v2)
 
static double L2squareNormDistance (const std::vector< float > &v1, const std::vector< float > &v2)
 
static cv::Mat vectortomat (const std::vector< float > &vector)
 

Constructor & Destructor Documentation

◆ DNNSFaceExtractor()

Digikam::DNNSFaceExtractor::DNNSFaceExtractor ( )

References loadModels().

◆ ~DNNSFaceExtractor()

Digikam::DNNSFaceExtractor::~DNNSFaceExtractor ( )
override

Member Function Documentation

◆ alignFace()

cv::Mat Digikam::DNNSFaceExtractor::alignFace ( const cv::Mat &  inputImage) const
overridevirtual

◆ cosineDistance()

double Digikam::DNNFaceExtractorBase::cosineDistance ( const std::vector< float > &  v1,
const std::vector< float > &  v2 
)
staticinherited

Calculate different between 2 vectors.

◆ decodeVector()

std::vector< float > Digikam::DNNFaceExtractorBase::decodeVector ( const QJsonArray &  json)
staticinherited

◆ encodeVector()

QJsonArray Digikam::DNNFaceExtractorBase::encodeVector ( const std::vector< float > &  vector)
staticinherited

◆ getFaceEmbedding()

cv::Mat Digikam::DNNSFaceExtractor::getFaceEmbedding ( const cv::Mat &  faceImage)
overridevirtual

◆ L2squareDistance()

double Digikam::DNNFaceExtractorBase::L2squareDistance ( const std::vector< float > &  v1,
const std::vector< float > &  v2 
)
staticinherited

◆ L2squareNormDistance()

double Digikam::DNNFaceExtractorBase::L2squareNormDistance ( const std::vector< float > &  v1,
const std::vector< float > &  v2 
)
staticinherited

◆ loadModels()

bool Digikam::DNNSFaceExtractor::loadModels ( )
overridevirtual

Read pretrained neural network for face recognition.

Implements Digikam::DNNFaceExtractorBase.

References Digikam::DNNModelManager::getModel(), and Digikam::DNNModelManager::instance().

Referenced by DNNSFaceExtractor().

◆ vectortomat()

cv::Mat Digikam::DNNFaceExtractorBase::vectortomat ( const std::vector< float > &  vector)
staticinherited

Convert face embedding between different formats.


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