17 class Fftw : public Fft
21 Fftw(int num_of_threads);
22 void init(unsigned width, unsigned height, unsigned num_of_feats, unsigned num_of_scales, bool big_batch_mode) override;
23 void set_window(const cv::Mat &window) override;
24 ComplexMat forward(const cv::Mat &input) override;
25 ComplexMat forward_raw(float *input, bool all_scales) override;
26 ComplexMat forward_window(const std::vector<cv::Mat> &input) override;
27 cv::Mat inverse(const ComplexMat &input) override;
28 float* inverse_raw(const ComplexMat &input) override;
31 unsigned m_num_threads = 6;
32 unsigned m_width, m_height, m_num_of_feats, m_num_of_scales;
33 bool m_big_batch_mode;
35 fftwf_plan plan_f, plan_f_all_scales, plan_fw, plan_fw_all_scales, plan_i_features,
36 plan_i_features_all_scales, plan_i_1ch, plan_i_1ch_all_scales;