Program Listing for File vision_lite.h

↰ Return to documentation for file (include/runtime/include/dataset/vision_lite.h)

#ifndef MINDSPORE_CCSRC_MINDDATA_DATASET_INCLUDE_DATASET_VISION_LITE_H_
#define MINDSPORE_CCSRC_MINDDATA_DATASET_INCLUDE_DATASET_VISION_LITE_H_

#include <map>
#include <memory>
#include <string>
#include <utility>
#include <vector>

#include "include/api/status.h"
#include "include/dataset/constants.h"
#include "include/dataset/transforms.h"

namespace mindspore {
namespace dataset {
// Transform operations for performing computer vision.
namespace vision {
// Forward Declarations
class RotateOperation;

class DATASET_API Affine final : public TensorTransform {
 public:
  explicit Affine(float_t degrees, const std::vector<float> &translation = {0.0, 0.0}, float scale = 0.0,
                  const std::vector<float> &shear = {0.0, 0.0},
                  InterpolationMode interpolation = InterpolationMode::kNearestNeighbour,
                  const std::vector<uint8_t> &fill_value = {0, 0, 0});

  ~Affine() = default;

  std::shared_ptr<TensorOperation> Parse() override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API CenterCrop final : public TensorTransform {
 public:
  explicit CenterCrop(const std::vector<int32_t> &size);

  ~CenterCrop() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;

  std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API Crop final : public TensorTransform {
 public:
  Crop(const std::vector<int32_t> &coordinates, const std::vector<int32_t> &size);

  ~Crop() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API Decode final : public TensorTransform {
 public:
  explicit Decode(bool rgb = true);

  ~Decode() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;

  std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API GaussianBlur final : public TensorTransform {
 public:
  explicit GaussianBlur(const std::vector<int32_t> &kernel_size, const std::vector<float> &sigma = {0., 0.});

  ~GaussianBlur() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API HWC2CHW final : public TensorTransform {
 public:
  HWC2CHW();

  ~HWC2CHW() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;
};

class DATASET_API Normalize final : public TensorTransform {
 public:

  Normalize(const std::vector<float> &mean, const std::vector<float> &std, bool is_hwc = true);

  ~Normalize() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;

  std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API Pad final : public TensorTransform {
 public:
  explicit Pad(const std::vector<int32_t> &padding, const std::vector<uint8_t> &fill_value = {0},
               BorderType padding_mode = BorderType::kConstant);

  ~Pad() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API RandomAffine final : public TensorTransform {
 public:
  explicit RandomAffine(const std::vector<float_t> &degrees,
                        const std::vector<float_t> &translate_range = {0.0, 0.0, 0.0, 0.0},
                        const std::vector<float_t> &scale_range = {1.0, 1.0},
                        const std::vector<float_t> &shear_ranges = {0.0, 0.0, 0.0, 0.0},
                        InterpolationMode interpolation = InterpolationMode::kNearestNeighbour,
                        const std::vector<uint8_t> &fill_value = {0, 0, 0});

  ~RandomAffine() = default;

  std::shared_ptr<TensorOperation> Parse() override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API Rescale final : public TensorTransform {
 public:
  Rescale(float rescale, float shift);

  ~Rescale() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API Resize final : public TensorTransform {
 public:
  explicit Resize(const std::vector<int32_t> &size, InterpolationMode interpolation = InterpolationMode::kLinear);

  ~Resize() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;

  std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API ResizePreserveAR final : public TensorTransform {
 public:
  ResizePreserveAR(int32_t height, int32_t width, int32_t img_orientation = 0);

  ~ResizePreserveAR() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;

 private:
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API RGB2BGR final : public TensorTransform {
 public:
  RGB2BGR() = default;

  ~RGB2BGR() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;
};

class DATASET_API RGB2GRAY final : public TensorTransform {
 public:
  RGB2GRAY() = default;

  ~RGB2GRAY() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;
};

class DATASET_API Rotate final : public TensorTransform {
 public:
  explicit Rotate(FixRotationAngle angle_id = FixRotationAngle::k0Degree);

  explicit Rotate(float degrees, InterpolationMode resample = InterpolationMode::kNearestNeighbour, bool expand = false,
                  const std::vector<float> &center = {}, const std::vector<uint8_t> &fill_value = {0, 0, 0});

  ~Rotate() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;

 private:
  std::shared_ptr<RotateOperation> op_;
  struct Data;
  std::shared_ptr<Data> data_;
};

class DATASET_API SwapRedBlue final : public TensorTransform {
 public:
  SwapRedBlue();

  ~SwapRedBlue() = default;

 protected:
  std::shared_ptr<TensorOperation> Parse() override;
};
}  // namespace vision
}  // namespace dataset
}  // namespace mindspore
#endif  // MINDSPORE_CCSRC_MINDDATA_DATASET_INCLUDE_DATASET_VISION_LITE_H_