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_