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
18
#include "
dnnbasedetectormodel.h
"
19
20
namespace
Digikam
21
{
22
23
class
DIGIKAM_GUI_EXPORT
DNNResnetDetector
:
public
DNNBaseDetectorModel
24
{
25
26
public
:
27
28
explicit
DNNResnetDetector
();
29
~DNNResnetDetector
()
override
=
default
;
30
31
protected
:
32
33
bool
loadModels()
override
;
34
35
private
:
36
37
QHash<QString, QVector<QRect> > postprocess(
const
cv::Mat& inputImage,
38
const
cv::Mat& out)
const override
;
39
40
private
:
41
42
// Disable
43
DNNResnetDetector
(
const
DNNResnetDetector
&) =
delete
;
44
DNNResnetDetector
& operator=(
const
DNNResnetDetector
&) =
delete
;
45
};
46
47
}
// namespace Digikam
Digikam::DNNBaseDetectorModel
Definition
dnnbasedetectormodel.h:40
Digikam::DNNResnetDetector
Definition
dnnresnetdetector.h:24
Digikam::DNNResnetDetector::~DNNResnetDetector
~DNNResnetDetector() override=default
dnnbasedetectormodel.h
Digikam
Definition
datefolderview.cpp:34
core
libs
tags
autoassignment
model
dnnresnetdetector.h
Generated by
1.9.8