]> rtime.felk.cvut.cz Git - hercules2020/kcf.git/blobdiff - src/fft.cpp
Do not use virtual methods in FFT class
[hercules2020/kcf.git] / src / fft.cpp
index 412618d070b864f72c8d4dee6e30ff2421aaefab..e6be093194333041eb0778fb89c357ea2d7376c9 100644 (file)
@@ -3,11 +3,6 @@
 #include <cassert>
 #include "debug.h"
 
-Fft::~Fft()
-{
-
-}
-
 void Fft::init(unsigned width, unsigned height, unsigned num_of_feats, unsigned num_of_scales)
 {
     m_width = width;
@@ -41,8 +36,8 @@ void Fft::forward(const MatScales &real_input, ComplexMat &complex_result)
     assert(real_input.size[1] == int(m_height));
     assert(real_input.size[2] == int(m_width));
 
-    assert(complex_result.cols = freq_size(cv::Size(m_width, m_height)).width);
-    assert(complex_result.rows = freq_size(cv::Size(m_width, m_height)).height);
+    assert(int(complex_result.cols) == freq_size(cv::Size(m_width, m_height)).width);
+    assert(int(complex_result.rows) == freq_size(cv::Size(m_width, m_height)).height);
     assert(complex_result.channels() == uint(real_input.size[0]));
 
     (void)real_input;
@@ -67,6 +62,10 @@ void Fft::forward_window(MatScaleFeats &patch_feats, ComplexMat &complex_result,
         assert(tmp.size[2] == patch_feats.size[2]);
         assert(tmp.size[3] == patch_feats.size[3]);
 
+        assert(int(complex_result.cols) == freq_size(cv::Size(m_width, m_height)).width);
+        assert(int(complex_result.rows) == freq_size(cv::Size(m_width, m_height)).height);
+        assert(complex_result.channels() == uint(patch_feats.size[0] * patch_feats.size[1]));
+
         (void)patch_feats;
         (void)complex_result;
         (void)tmp;
@@ -85,6 +84,10 @@ void Fft::inverse(ComplexMat &complex_input, MatScales &real_result)
     assert(real_result.size[1] == int(m_height));
     assert(real_result.size[2] == int(m_width));
 
+    assert(int(complex_input.cols) == freq_size(cv::Size(m_width, m_height)).width);
+    assert(int(complex_input.rows) == freq_size(cv::Size(m_width, m_height)).height);
+    assert(complex_input.channels() == uint(real_result.size[0]));
+
     (void)complex_input;
     (void)real_result;
 }