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
22
#include "
dnnfaceextractorbase.h
"
23
#include "
dnnmodelyunet.h
"
24
25
namespace
Digikam
26
{
27
28
// TODO: remove export macro after testing
29
30
class
DIGIKAM_GUI_EXPORT
DNNSFaceExtractor
:
public
DNNFaceExtractorBase
31
{
32
33
public
:
34
35
DNNSFaceExtractor
();
36
~DNNSFaceExtractor
()
override
;
37
38
public
:
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
48
private
:
49
51
DNNSFaceExtractor
(
const
DNNSFaceExtractor
&) =
delete
;
52
DNNSFaceExtractor
& operator=(
const
DNNSFaceExtractor
&) =
delete
;
53
54
private
:
55
56
class
Private;
57
Private*
const
d =
nullptr
;
58
};
59
60
}
// namespace Digikam
Digikam::DNNFaceExtractorBase
Definition
dnnfaceextractorbase.h:38
Digikam::DNNSFaceExtractor
Definition
dnnsfaceextractor.h:31
dnnfaceextractorbase.h
dnnmodelyunet.h
Digikam
Definition
datefolderview.cpp:34
core
libs
facesengine
recognition
opencv-dnn
dnnsfaceextractor.h
Generated by
1.9.8