OnnxSourceTask.h

Go to the documentation of this file.

#ifndef _SEIMPLEMENTATION_PLUGIN_ONNXSOURCETASK_H_
#define _SEIMPLEMENTATION_PLUGIN_ONNXSOURCETASK_H_

#include "SEFramework/Task/SourceTask.h"
#include "SEImplementation/Common/OnnxModel.h"

namespace SourceXtractor {

class OnnxSourceTask: public SourceTask {
public:
  struct OnnxModelInfo {
    std::shared_ptr<OnnxModel> model;
    std::string prop_name;
  };

  explicit OnnxSourceTask(const std::vector<OnnxModelInfo>& model_infos);

  ~OnnxSourceTask() override = default;

  void computeProperties(SourceInterface& source) const override;

private:

  const std::vector<OnnxModelInfo>& m_model_infos;
};

} // end of namespace SourceXtractor

#endif // _SEIMPLEMENTATION_PLUGIN_ONNXSOURCETASK_H_