src/fft_opencv.h

   1
   2 #ifndef FFTOPENCV_H
   3 #define FFTOPENCV_H
   4
   5 #include "fft.h"
   6
   7 struct Scale_vars;
   8
   9 class FftOpencv : public Fft
  10 {
  11 public:
  12     void init(unsigned width, unsigned height, unsigned num_of_feats, unsigned num_of_scales, bool big_batch_mode) override;
  13     void set_window(const cv::Mat & window) override;
  14     void forward(Scale_vars & vars) override;
  15     void forward_window(Scale_vars & vars) override;
  16     void inverse(Scale_vars & vars) override;
  17     ~FftOpencv() override;
  18 private:
  19     cv::Mat m_window;
  20 };
  21
  22 #endif // FFTOPENCV_H