src/fft_opencv.h

   1
   2 #ifndef FFTOPENCV_H
   3 #define FFTOPENCV_H
   4
   5 #include "fft.h"
   6
   7 class FftOpencv : public Fft
   8 {
   9 public:
  10     void init(unsigned width, unsigned height, unsigned num_of_feats, unsigned num_of_scales) override;
  11     void set_window(const cv::Mat & window) override;
  12     void forward(const cv::Mat & real_input, ComplexMat & complex_result, float *real_input_arr, cudaStream_t  stream) override;
  13     void forward_window(std::vector<cv::Mat> patch_feats, ComplexMat & complex_result, cv::Mat & fw_all, float *real_input_arr, cudaStream_t stream) override;
  14     void inverse(ComplexMat &  complex_input, cv::Mat & real_result, float *real_result_arr, cudaStream_t stream) override;
  15     ~FftOpencv() override;
  16 private:
  17     cv::Mat m_window;
  18 };
  19
  20 #endif // FFTOPENCV_H