Add convenience ComplexMat constructors + code updates

[hercules2020/kcf.git] / src / complexmat.cuh

diff --git a/src/complexmat.cuh b/src/complexmat.cuh
index 907ea1f55d9f7c1ebd0af9ff83764ca4af4e713d..940900e7aa141281f898348238eeeae097acfa5f 100644 (file)
--- a/src/complexmat.cuh
+++ b/src/complexmat.cuh
@@ -30,6 +30,12 @@ class ComplexMat {
         CudaSafeCall(cudaMalloc(&p_data, n_channels * cols * rows * sizeof(cufftComplex)));
     }
 
+    ComplexMat(cv::Size size, uint _n_channels)
+        : cols(size.width), rows(size.height), n_channels(_n_channels)
+    {
+        CudaSafeCall(cudaMalloc(&p_data, n_channels * cols * rows * sizeof(cufftComplex)));
+    }
+
     ComplexMat(ComplexMat &&other)
     {
         cols = other.cols;