digiKam
Loading...
Searching...
No Matches
dnnresnetdetector.h
Go to the documentation of this file.
1/* ============================================================
2 *
3 * This file is a part of digiKam
4 * https://www.digikam.org
5 *
6 * Date : 2023-09-02
7 * Description : Derived class to perform Resnet neural network inference
8 * for object detection (including yolo versions to benchmark).
9 *
10 * SPDX-FileCopyrightText: 2023 by Quoc Hung TRAN <quochungtran1999 at gmail dot com>
11 *
12 * SPDX-License-Identifier: GPL-2.0-or-later
13 *
14 * ============================================================ */
15
16#pragma once
17
19
20namespace Digikam
21{
22
23class DIGIKAM_GUI_EXPORT DNNResnetDetector: public DNNBaseDetectorModel
24{
25
26public:
27
28 explicit DNNResnetDetector();
29 ~DNNResnetDetector() override = default;
30
31protected:
32
33 bool loadModels() override;
34
35private:
36
37 QHash<QString, QVector<QRect> > postprocess(const cv::Mat& inputImage,
38 const cv::Mat& out) const override;
39
40private:
41
42 // Disable
43 DNNResnetDetector(const DNNResnetDetector&) = delete;
44 DNNResnetDetector& operator=(const DNNResnetDetector&) = delete;
45};
46
47} // namespace Digikam
Definition dnnbasedetectormodel.h:40
Definition dnnresnetdetector.h:24
~DNNResnetDetector() override=default
Definition datefolderview.cpp:34