Do not use virtual methods in FFT class

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

diff --git a/src/fft_opencv.h b/src/fft_opencv.h
index 52c107554198b3dd6bc19ccffef376f0e49d61b2..032989b422b0148b47464b6db53f5b3de5783e6d 100644 (file)
--- a/src/fft_opencv.h
+++ b/src/fft_opencv.h
@@ -7,12 +7,12 @@
 class FftOpencv : 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(const cv::Mat & real_input, ComplexMat & complex_result, float *real_input_arr, cudaStream_t  stream) override;
-    void forward_window(std::vector<cv::Mat> patch_feats, ComplexMat & complex_result, cv::Mat & fw_all, float *real_input_arr, cudaStream_t stream) override;
-    void inverse(ComplexMat &  complex_input, cv::Mat & real_result, float *real_result_arr, cudaStream_t stream) override;
-    ~FftOpencv() override;
+    void init(unsigned width, unsigned height, unsigned num_of_feats, unsigned num_of_scales);
+    void set_window(const MatDynMem &window);
+    void forward(const MatScales &real_input, ComplexMat &complex_result);
+    void forward_window(MatScaleFeats &patch_feats_in, ComplexMat &complex_result, MatScaleFeats &tmp);
+    void inverse(ComplexMat &complex_input, MatScales &real_result);
+    ~FftOpencv();
 private:
     cv::Mat m_window;
 };