digiKam
Loading...
Searching...
No Matches
dnnsfaceextractor.h
Go to the documentation of this file.
1/* ============================================================
2 *
3 * This file is a part of digiKam
4 *
5 * Date : 2019-06-01
6 * Description : Face recognition using deep learning.
7 * The internal DNN library interface.
8 *
9 * SPDX-FileCopyrightText: 2019 by Thanh Trung Dinh <dinhthanhtrung1996 at gmail dot com>
10 * SPDX-FileCopyrightText: 2020-2024 by Gilles Caulier <caulier dot gilles at gmail dot com>
11 * SPDX-FileCopyrightText: 2020 by Nghia Duong <minhnghiaduong997 at gmail dot com>
12 * SPDX-FileCopyrightText: 2024 by Michael Miller <michael underscore miller at msn dot com>
13 *
14 * SPDX-License-Identifier: GPL-2.0-or-later
15 *
16 * ============================================================ */
17
18#pragma once
19
20// Local includes
21
23#include "dnnmodelyunet.h"
24
25namespace Digikam
26{
27
28// TODO: remove export macro after testing
29
30class DIGIKAM_GUI_EXPORT DNNSFaceExtractor: public DNNFaceExtractorBase
31{
32
33public:
34
36 ~DNNSFaceExtractor() override;
37
38public:
39
43 bool loadModels() override;
44
45 virtual cv::Mat alignFace(const cv::Mat& inputImage) const override;
46 virtual cv::Mat getFaceEmbedding(const cv::Mat& faceImage) override;
47
48private:
49
51 DNNSFaceExtractor(const DNNSFaceExtractor&) = delete;
52 DNNSFaceExtractor& operator=(const DNNSFaceExtractor&) = delete;
53
54private:
55
56 class Private;
57 Private* const d = nullptr;
58};
59
60} // namespace Digikam
Definition dnnfaceextractorbase.h:38
Definition dnnsfaceextractor.h:31
Definition datefolderview.cpp:34