OnnxTaskFactory.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_PLUGIN_ONNXTASKFACTORY_H_
#define _SEIMPLEMENTATION_PLUGIN_ONNXTASKFACTORY_H_
 
#include "SEFramework/Task/TaskFactory.h"
#include "SEImplementation/Common/OnnxModel.h"
#include "SEImplementation/Plugin/Onnx/OnnxSourceTask.h"
 
namespace SourceXtractor {

class OnnxTaskFactory : public TaskFactory {
public:
  OnnxTaskFactory();
 
  virtual ~OnnxTaskFactory() = default;

  std::shared_ptr<Task> createTask(const PropertyId& property_id) const override;

  void reportConfigDependencies(Euclid::Configuration::ConfigManager& manager) const override;

  void configure(Euclid::Configuration::ConfigManager& manager) override;

  void registerPropertyInstances(OutputRegistry& registry) override;
 
private:
  std::vector<OnnxSourceTask::OnnxModelInfo> m_model_infos;
};
 
} // end of namespace SourceXtractor
 
 
#endif // _SEIMPLEMENTATION_PLUGIN_ONNXTASKFACTORY_H_