SourceXtractorPlusPlus
0.16
Please provide a description of the project.
SEImplementation
SEImplementation
Plugin
Onnx
OnnxSourceTask.h
Go to the documentation of this file.
1
18
#ifndef _SEIMPLEMENTATION_PLUGIN_ONNXSOURCETASK_H_
19
#define _SEIMPLEMENTATION_PLUGIN_ONNXSOURCETASK_H_
20
21
#include "
SEFramework/Task/SourceTask.h
"
22
#include "
SEImplementation/Common/OnnxModel.h
"
23
24
namespace
SourceXtractor
{
25
29
class
OnnxSourceTask
:
public
SourceTask
{
30
public
:
31
struct
OnnxModelInfo
{
32
std::shared_ptr<OnnxModel>
model
;
33
std::string
prop_name
;
34
};
35
41
explicit
OnnxSourceTask
(
const
std::vector<OnnxModelInfo>
& model_infos);
42
46
~OnnxSourceTask
()
override
=
default
;
47
49
void
computeProperties
(
SourceInterface
& source)
const override
;
50
51
private
:
52
53
const
std::vector<OnnxModelInfo>
&
m_model_infos
;
54
};
55
56
}
// end of namespace SourceXtractor
57
58
#endif
// _SEIMPLEMENTATION_PLUGIN_ONNXSOURCETASK_H_
OnnxModel.h
SourceTask.h
std::string
SourceXtractor::OnnxSourceTask
Definition:
OnnxSourceTask.h:29
SourceXtractor::OnnxSourceTask::computeProperties
void computeProperties(SourceInterface &source) const override
Computes one or more properties for the Source.
Definition:
OnnxSourceTask.cpp:93
SourceXtractor::OnnxSourceTask::OnnxSourceTask
OnnxSourceTask(const std::vector< OnnxModelInfo > &model_infos)
Definition:
OnnxSourceTask.cpp:50
SourceXtractor::OnnxSourceTask::~OnnxSourceTask
~OnnxSourceTask() override=default
SourceXtractor::OnnxSourceTask::m_model_infos
const std::vector< OnnxModelInfo > & m_model_infos
Definition:
OnnxSourceTask.h:53
SourceXtractor::SourceInterface
The SourceInterface is an abstract "source" that has properties attached to it.
Definition:
SourceInterface.h:46
SourceXtractor::SourceTask
A Task that acts on a Source to compute one or more properties.
Definition:
SourceTask.h:36
SourceXtractor
Definition:
Aperture.h:30
std::shared_ptr
SourceXtractor::OnnxSourceTask::OnnxModelInfo
Definition:
OnnxSourceTask.h:31
SourceXtractor::OnnxSourceTask::OnnxModelInfo::model
std::shared_ptr< OnnxModel > model
Definition:
OnnxSourceTask.h:32
SourceXtractor::OnnxSourceTask::OnnxModelInfo::prop_name
std::string prop_name
Definition:
OnnxSourceTask.h:33
std::vector
Generated by
1.9.1