image_transforms.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
 
#include "pulsatrix/device_backend.hpp"
#include "pulsatrix/transform.hpp"
 
namespace pulsatrix {
 
class ResizeTransform : public Transform {
public:
    ResizeTransform(int64_t target_height, int64_t target_width, DeviceBackend* backend)
        : target_height_(target_height), target_width_(target_width), backend_(backend) {}
 
    [[nodiscard]] Sample apply(Sample sample) const override;
 
private:
    int64_t target_height_;
    int64_t target_width_;
    DeviceBackend* backend_;
};
 
class CenterCropTransform : public Transform {
public:
    CenterCropTransform(int64_t crop_height, int64_t crop_width, DeviceBackend* backend)
        : crop_height_(crop_height), crop_width_(crop_width), backend_(backend) {}
 
    [[nodiscard]] Sample apply(Sample sample) const override;
 
private:
    int64_t crop_height_;
    int64_t crop_width_;
    DeviceBackend* backend_;
};
 
class NormalizeTransform : public Transform {
public:
    NormalizeTransform(std::vector<float> mean, std::vector<float> std) : mean_(std::move(mean)), std_(std::move(std)) {}
 
    [[nodiscard]] Sample apply(Sample sample) const override;
 
private:
    std::vector<float> mean_;
    std::vector<float> std_;
};
 
class HorizontalFlipTransform : public Transform {
public:
    [[nodiscard]] Sample apply(Sample sample) const override;
};
 
}  // namespace pulsatrix