write_json_op.h 609 字节
Newer Older
W
wangguibao 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
#ifndef BAIDU_PADDLE_SERVING_PREDICTOR_OP_WRITE_JSON_OP_H
#define BAIDU_PADDLE_SERVING_PREDICTOR_OP_WRITE_JSON_OP_H

#include "builtin_format.pb.h"
#include "image_classification.pb.h"
#include "common/inner_common.h"
#include "op/op.h"
#include "framework/channel.h"
#include "framework/op_repository.h"

namespace baidu {
namespace paddle_serving {
namespace predictor {

class WriteJsonOp : public OpWithChannel<
      baidu::paddle_serving::predictor::image_classification::Response> {
public: 

    DECLARE_OP(WriteJsonOp);

    int inference();
};

} // predictor
} // paddle_serving
} // baidu

#endif