norm.hpp

Go to the documentation of this file.

#pragma once
namespace reusex::vision::tensor_rt::norm_image {
 
// cppcheck-suppress syntaxError
enum class NormType : int { None = 0, MeanStd = 1, AlphaBeta = 2 };
 
// cppcheck-suppress syntaxError
enum class ChannelType : int { None = 0, SwapRB = 1 };
 
struct Norm {
  float mean[3] = {};
  float std[3] = {};
  float alpha = 0.f;
  float beta = 0.f;
  NormType type = NormType::None;
  ChannelType channel_type = ChannelType::None;
 
  // out = (x * alpha - mean) / std
  static Norm mean_std(const float mean[3], const float std[3],
                       float alpha = 1 / 255.0f,
                       ChannelType channel_type = ChannelType::None);
 
  // out = x * alpha + beta
  static Norm alpha_beta(float alpha, float beta = 0,
                         ChannelType channel_type = ChannelType::None);
 
  // None
  static Norm None();
};
 
} // namespace reusex::vision::tensor_rt::norm_image