digiKam
Loading...
Searching...
No Matches
dnnfacedetectorssd.h
Go to the documentation of this file.
1
/* ============================================================
2
*
3
* This file is a part of digiKam
4
*
5
* Date : 2019-08-08
6
* Description : Derived class to perform SSD neural network inference
7
* for face detection
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
*
12
* SPDX-License-Identifier: GPL-2.0-or-later
13
*
14
* ============================================================ */
15
16
#pragma once
17
18
// Local includes
19
20
#include "
dnnfacedetectorbase.h
"
21
22
namespace
Digikam
23
{
24
25
class
DIGIKAM_EXPORT
DNNFaceDetectorSSD
:
public
DNNFaceDetectorBase
26
{
27
28
public
:
29
30
explicit
DNNFaceDetectorSSD
();
31
~DNNFaceDetectorSSD
()
override
=
default
;
32
33
bool
loadModels();
34
35
void
detectFaces(
const
cv::Mat& inputImage,
36
const
cv::Size& paddedSize,
37
std::vector<cv::Rect>& detectedBboxes)
override
;
38
39
private
:
40
41
void
postprocess(cv::Mat detectionMat,
42
const
cv::Size& paddedSize,
43
std::vector<cv::Rect>& detectedBboxes)
const
;
44
45
private
:
46
47
// Disable
48
DNNFaceDetectorSSD
(
const
DNNFaceDetectorSSD
&) =
delete
;
49
DNNFaceDetectorSSD
& operator=(
const
DNNFaceDetectorSSD
&) =
delete
;
50
};
51
52
}
// namespace Digikam
Digikam::DNNFaceDetectorBase
Definition
dnnfacedetectorbase.h:37
Digikam::DNNFaceDetectorSSD
Definition
dnnfacedetectorssd.h:26
Digikam::DNNFaceDetectorSSD::~DNNFaceDetectorSSD
~DNNFaceDetectorSSD() override=default
dnnfacedetectorbase.h
Digikam
Definition
datefolderview.cpp:34
core
libs
facesengine
detection
opencv-dnn
dnnfacedetectorssd.h
Generated by
1.9.8