FFT does not take whole Scale_vars struct

[hercules2020/kcf.git] / src / fft_cufft.h

diff --git a/src/fft_cufft.h b/src/fft_cufft.h
index 5c20762d9e0d921320ba9e8aded59cf41b4d17b8..48a29800504b57e4fce662b4c35bfe63e1f7ae4c 100644 (file)
--- a/src/fft_cufft.h
+++ b/src/fft_cufft.h
@@ -23,9 +23,9 @@ class cuFFT : public Fft
 public:
     void init(unsigned width, unsigned height, unsigned num_of_feats, unsigned num_of_scales, bool big_batch_mode) override;
     void set_window(const cv::Mat & window) override;
-    void forward(Scale_vars & vars) override;
-    void forward_window(Scale_vars & vars) override;
-    void inverse(Scale_vars & vars) override;
+    void forward(const cv::Mat & real_input, ComplexMat & complex_result, float *real_input_arr) override;
+    void forward_window(std::vector<cv::Mat> patch_feats, ComplexMat & complex_result, cv::Mat & fw_all, float *real_input_arr) override;
+    void inverse(ComplexMat &  complex_input, cv::Mat & real_result, float *real_result_arr) override;
     ~cuFFT() override;
 private:
     cv::Mat m_window;