digiKam
Loading...
Searching...
No Matches
dnnmodelyunet.h
Go to the documentation of this file.
1/* ============================================================
2 *
3 * This file is a part of digiKam project
4 * https://www.digikam.org
5 *
6 * Date : 2024-10-13
7 * Description : digiKam DNNModelYuNet specific for create FaceDetectorYN instances
8 *
9 * SPDX-FileCopyrightText: 2024 by Michael Miller <michael underscore miller at msn dot com>
10 *
11 * SPDX-License-Identifier: GPL-2.0-or-later
12 *
13 * ============================================================ */
14
15#pragma once
16
17// Qt includes
18
19#include <QMutex>
20
21// Local includes
22
23#include "digikam_opencv.h"
24#include "digikam_export.h"
25#include "dnnmodelmanager.h"
26#include "dnnmodelbase.h"
27
28namespace Digikam
29{
30
31class DIGIKAM_EXPORT DNNModelYuNet: public DNNModelBase
32{
33
34public:
35
36 // ---------- public methods ----------
37
38 explicit DNNModelYuNet(const DNNModelInfoContainer& _info);
39 virtual ~DNNModelYuNet() override = default;
40
41public:
42
43 // ---------- public members ----------
44
45 cv::Ptr<cv::FaceDetectorYN>& getNet();
46
47private:
48
49 cv::Ptr<cv::FaceDetectorYN> net;
50
51private:
52
53 DNNModelYuNet() = delete;
54
55 virtual bool loadModel() override;
56 bool callLoader();
57};
58
59} // namespace Digikam
Definition dnnmodelbase.h:35
Definition dnnmodelinfocontainer.h:33
Definition dnnmodelyunet.h:32
virtual ~DNNModelYuNet() override=default
Definition datefolderview.cpp:34