![]() |
digiKam Developer Documentation
Professional Photo Management with the Power of Open Source
|
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 override |
| virtual int | predict (const cv::UMat &target) 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 |
|
explicit |
|
inlineoverridevirtual |
Implements Digikam::MLClassifierFoundation.
|
inlineoverridevirtual |
Implements Digikam::MLClassifierFoundation.
|
overridevirtual |
Reimplemented from Digikam::AutotagsClassifierBase.
|
inlineoverridevirtual |
Reimplemented from Digikam::AutotagsClassifierBase.
| void Digikam::AutotagsClassifierYolo::setParams | ( | YoloVersion | version, |
| QSize | size | ||
| ) |