SourceXtractorPlusPlus
0.16
Please provide a description of the project.
|
#include <OnnxPlugin.h>
Public Member Functions | |
virtual | ~OnnxPlugin ()=default |
Destructor. More... | |
std::string | getIdString () const override |
void | registerPlugin (PluginAPI &plugin_api) override |
![]() | |
virtual | ~Plugin ()=default |
The ONNX plugin wraps a machine learning model stored using the "Open Neural Network Exchange" format. SourceXtractor will feed the model a stamp with the size the model requires as an input. For the output column, the 'domain' field + the graph name from the ONNX model will be used: i.e. ch.unige.astro_galaxy_class
Definition at line 35 of file OnnxPlugin.h.
|
virtualdefault |
Destructor.
|
overridevirtual |
Implements SourceXtractor::Plugin.
Definition at line 28 of file OnnxPlugin.cpp.
|
overridevirtual |
Implements SourceXtractor::Plugin.
Definition at line 32 of file OnnxPlugin.cpp.
References SourceXtractor::OutputRegistry::enableOutput(), SourceXtractor::PluginAPI::getOutputRegistry(), SourceXtractor::PluginAPI::getTaskFactoryRegistry(), and SourceXtractor::TaskFactoryRegistry::registerTaskFactory().