![]() |
digiKam Developer Documentation
Professional Photo Management with the Power of Open Source
|
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) |
![]() | |
AutotagsClassifierBase (float _threshold, const QString &_predefinedClassFile) | |
QList< QString > | getClassStrings (const QList< int > &labelList) const |
bool | retrain () override |
![]() | |
void | setThreshold (float _threshold) |
Additional Inherited Members | |
![]() | |
bool | loadTrainingData () override |
![]() | |
QList< QString > | predefinedClasses |
QString | predefinedClassFile |
![]() | |
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 | ||
) |