SourceXtractorPlusPlus  0.16
Please provide a description of the project.
Public Member Functions | List of all members
SourceXtractor::OnnxPlugin Class Reference

#include <OnnxPlugin.h>

Inheritance diagram for SourceXtractor::OnnxPlugin:
[legend]
Collaboration diagram for SourceXtractor::OnnxPlugin:
[legend]

Public Member Functions

virtual ~OnnxPlugin ()=default
 Destructor. More...
 
std::string getIdString () const override
 
void registerPlugin (PluginAPI &plugin_api) override
 
- Public Member Functions inherited from SourceXtractor::Plugin
virtual ~Plugin ()=default
 

Detailed Description

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

See also
https://onnx.ai/

Definition at line 35 of file OnnxPlugin.h.

Constructor & Destructor Documentation

◆ ~OnnxPlugin()

virtual SourceXtractor::OnnxPlugin::~OnnxPlugin ( )
virtualdefault

Destructor.

Member Function Documentation

◆ getIdString()

std::string SourceXtractor::OnnxPlugin::getIdString ( ) const
overridevirtual

Implements SourceXtractor::Plugin.

Definition at line 28 of file OnnxPlugin.cpp.

◆ registerPlugin()

void SourceXtractor::OnnxPlugin::registerPlugin ( PluginAPI plugin_api)
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().

Here is the call graph for this function:

The documentation for this class was generated from the following files: