Method Setup

include/singa/io/transformer.h:43–55

class ImageTransformer: public Transformer {
 public:
  void Setup(const TransformerConf& conf) override {
    featurewise_center_ = conf.featurewise_center();
    featurewise_std_norm_ = conf.featurewise_std_norm();
    resize_height_ = conf.resize_height();
    resize_width_ = conf.resize_width();
    rescale_ = conf.rescale();
    horizontal_mirror_ = conf.horizontal_mirror();
    image_dim_order_ = conf.image_dim_order();
    
    /// if crop_shape not contain 2 elements, ignore crop option.
    if (conf.crop_shape_size() == 2)
      crop_shape_ = {conf.crop_shape(0), conf.crop_shape(1)};      
  }

  Tensor Apply(int flag, Tensor& input) override;