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

#include <autotagsassign.h>

Public Member Functions

 AutoTagsAssign (DetectorModel model=DetectorModel::YOLOV5NANO)
 
QList< QString > generateTagsList (const DImg &inputImage)
 
QList< QString > generateTagsList (const QImage &inputImage)
 
QList< QList< QString > > generateTagsList (const QList< DImg > &inputImages, int batchSize) const
 
QList< QList< QString > > generateTagsList (const QList< QString > &inputImagePaths, int batchSize) const
 
QList< QString > generateTagsList (const QString &inputImagePath)
 
QList< QString > getPredefinedTagsPath () const
 
cv::Mat prepareForDetection (const DImg &inputImage) const
 
cv::Mat prepareForDetection (const QImage &inputImage) const
 
std::vector< cv::Mat > prepareForDetection (const QList< DImg > &inputImages, int batchSize) const
 
std::vector< cv::Mat > prepareForDetection (const QList< QString > &inputImagePaths, int batchSize) const
 
cv::Mat prepareForDetection (const QString &inputImagePath) const
 
 ~AutoTagsAssign ()
 

Constructor & Destructor Documentation

◆ AutoTagsAssign()

Digikam::AutoTagsAssign::AutoTagsAssign ( DetectorModel  model = DetectorModel::YOLOV5NANO)
explicit

◆ ~AutoTagsAssign()

Digikam::AutoTagsAssign::~AutoTagsAssign ( )

Member Function Documentation

◆ generateTagsList() [1/5]

QList< QString > Digikam::AutoTagsAssign::generateTagsList ( const DImg inputImage)

◆ generateTagsList() [2/5]

QList< QString > Digikam::AutoTagsAssign::generateTagsList ( const QImage &  inputImage)

◆ generateTagsList() [3/5]

QList< QList< QString > > Digikam::AutoTagsAssign::generateTagsList ( const QList< DImg > &  inputImages,
int  batchSize 
) const

Run in batch return the list of tags name corresponding to. NOTE: the batch size is fixed depending on the deep NN model we choose.

References Digikam::DNNBaseDetectorModel::generateObjects(), and prepareForDetection().

◆ generateTagsList() [4/5]

QList< QList< QString > > Digikam::AutoTagsAssign::generateTagsList ( const QList< QString > &  inputImagePaths,
int  batchSize 
) const

◆ generateTagsList() [5/5]

QList< QString > Digikam::AutoTagsAssign::generateTagsList ( const QString &  inputImagePath)

◆ getPredefinedTagsPath()

QList< QString > Digikam::AutoTagsAssign::getPredefinedTagsPath ( ) const

◆ prepareForDetection() [1/5]

◆ prepareForDetection() [2/5]

cv::Mat Digikam::AutoTagsAssign::prepareForDetection ( const QImage &  inputImage) const

◆ prepareForDetection() [3/5]

std::vector< cv::Mat > Digikam::AutoTagsAssign::prepareForDetection ( const QList< DImg > &  inputImages,
int  batchSize 
) const

◆ prepareForDetection() [4/5]

std::vector< cv::Mat > Digikam::AutoTagsAssign::prepareForDetection ( const QList< QString > &  inputImagePaths,
int  batchSize 
) const

◆ prepareForDetection() [5/5]

cv::Mat Digikam::AutoTagsAssign::prepareForDetection ( const QString &  inputImagePath) const

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