| digiKam Developer Documentation Professional Photo Management with the Power of Open Source

Loading...
Searching...
No Matches
Public Types | Public Member Functions | List of all members
Digikam::AutotagsClassifierYolo Class Reference
Inheritance diagram for Digikam::AutotagsClassifierYolo:

## Public Types
enum | YoloVersion { YOLOv5 , YOLOv6 , YOLOv7 , YOLOv8 , YOLOv9 , YOLOv10 , YOLOv11 }

## Public Member Functions
| AutotagsClassifierYolo (float _threshold, const QString &predefinedClassFile)
virtual int | predict (const cv::Mat &target, const QList< int > &exclusionLabelList=QList< int >()) const override
virtual int | predict (const cv::UMat &target, const QList< int > &exclusionLabelList=QList< int >()) const override
virtual QList< int > | predictMulti (const QList< cv::Mat > &targets) const override
virtual QList< int > | predictMulti (const QList< cv::UMat > &targets) const override
void | setParams (YoloVersion version, QSize size)
Public Member Functions inherited from Digikam::AutotagsClassifierBase
| AutotagsClassifierBase (float _threshold, const QString &_predefinedClassFile)
QList< QString > | getClassStrings (const QList< int > &labelList) const
bool | retrain () override
Public Member Functions inherited from Digikam::MLClassifierFoundation
void | setThreshold (float _threshold)

## Additional Inherited Members
Protected Member Functions inherited from Digikam::AutotagsClassifierBase
bool | loadTrainingData () override
Protected Attributes inherited from Digikam::AutotagsClassifierBase
QList< QString > | predefinedClasses
QString | predefinedClassFile
Protected Attributes inherited from Digikam::MLClassifierFoundation
QReadWriteLock | lock
float | threshold = 0.0F

## Constructor & Destructor Documentation

## ◆ AutotagsClassifierYolo()

Digikam::AutotagsClassifierYolo::AutotagsClassifierYolo | ( | float | _threshold,
|  | const QString & | predefinedClassFile
| ) |  |
| explicit

Note
  : In strict technical terms, this is not a classifier. This is a post-processor. The initial classification is done by the YOLO detector, and this class is used to interpret the results of that classification and provide a list of labels based on the confidence of the classification.

## Member Function Documentation

## ◆ predict() [1/2]

virtual int Digikam::AutotagsClassifierYolo::predict | ( | const cv::Mat & | target,
|  | const QList< int > & | exclusionLabelList = `QList<int>()`
| ) |  | const
| inline override virtual

Implements Digikam::MLClassifierFoundation.

## ◆ predict() [2/2]

virtual int Digikam::AutotagsClassifierYolo::predict | ( | const cv::UMat & | target,
|  | const QList< int > & | exclusionLabelList = `QList<int>()`
| ) |  | const
| inline override virtual

Implements Digikam::MLClassifierFoundation.

## ◆ predictMulti() [1/2]

QList< int > Digikam::AutotagsClassifierYolo::predictMulti | ( | const QList< cv::Mat > & | targets | ) | const
| override virtual

Warning
  : This classifier does not handle YOLOv10

Reimplemented from Digikam::AutotagsClassifierBase.

## ◆ predictMulti() [2/2]

virtual QList< int > Digikam::AutotagsClassifierYolo::predictMulti | ( | const QList< cv::UMat > & | targets | ) | const
| inline override virtual

Reimplemented from Digikam::AutotagsClassifierBase.

## ◆ setParams()

void Digikam::AutotagsClassifierYolo::setParams | ( | YoloVersion | version,
|  | QSize | size
| ) |  |

Warning
  : This classifier does not handle YOLO v10