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
22namespace Digikam
23{
24
25class DIGIKAM_EXPORT DNNFaceDetectorSSD: public DNNFaceDetectorBase
26{
27
28public:
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
39private:
40
41 void postprocess(cv::Mat detectionMat,
42 const cv::Size& paddedSize,
43 std::vector<cv::Rect>& detectedBboxes) const;
44
45private:
46
47 // Disable
49 DNNFaceDetectorSSD& operator=(const DNNFaceDetectorSSD&) = delete;
50};
51
52} // namespace Digikam
Definition dnnfacedetectorbase.h:37
Definition dnnfacedetectorssd.h:26
~DNNFaceDetectorSSD() override=default
Definition datefolderview.cpp:34