generated from zhangwei/Matrixai
43 lines
763 B
C++
43 lines
763 B
C++
/**
|
|
* 数据处理引擎
|
|
**/
|
|
|
|
#ifndef TRAINSTEP1DATAREADENGINE_H
|
|
#define TRAINSTEP1DATAREADENGINE_H
|
|
|
|
#include "AppCommon.h"
|
|
|
|
#include "EngineBase.h"
|
|
#include "EngineFactory.h"
|
|
#include "json/json.h"
|
|
|
|
class TrainStep1DataReadEngine : public ai_matrix::EngineBase
|
|
{
|
|
public:
|
|
TrainStep1DataReadEngine();
|
|
~TrainStep1DataReadEngine();
|
|
|
|
APP_ERROR Init() override;
|
|
APP_ERROR DeInit() override;
|
|
APP_ERROR Process() override;
|
|
|
|
private:
|
|
|
|
bool readJson(std::string &strFilePath, Json::Value &jvInfo, int i = 1);
|
|
|
|
|
|
private:
|
|
std::string strPort0_;
|
|
std::string strPort1_;
|
|
std::string strPort2_;
|
|
|
|
ai_matrix::BaseConfig baseConfig_;
|
|
|
|
ai_matrix::DataSourceConfig dataSourceConfig_;
|
|
|
|
};
|
|
|
|
ENGINE_REGIST(TrainStep1DataReadEngine)
|
|
|
|
#endif
|