decoder.h

#ifndef SINGA_IO_DECODER_H_
#define SINGA_IO_DECODER_H_

#include <vector>
#include <string>
#include "singa/core/tensor.h"
#include "singa/proto/io.pb.h"

namespace singa {
class Decoder {
 public:
  Decoder() { }
  virtual ~Decoder() { }

  virtual void Setup(const DecoderConf& conf) {}

  virtual std::vector<Tensor> Decode(std::string value) = 0;
};

#ifdef USE_OPENCV
class JPGDecoder : public Decoder {
 public:
  void Setup(const DecoderConf& conf) override {
    image_dim_order_ = conf.image_dim_order();
  }
  std::vector<Tensor> Decode(std::string value) override;

  const std::string image_dim_order() const { return image_dim_order_; }

 private:
  std::string image_dim_order_ = "CHW";
};
#endif

class CSVDecoder : public Decoder {
 public:
  void Setup(const DecoderConf& conf) override {
    has_label_ = conf.has_label();
  }
  std::vector<Tensor> Decode(std::string value) override;

  const bool has_label() const { return has_label_; }

 private:
  bool has_label_ = false;
};
} // namespace singa
#endif // SINGA_IO_DECODER_H_