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_