]> rtime.felk.cvut.cz Git - hercules2020/kcf.git/commitdiff
Corrected warnings and errors in CUFFT version
authorShanigen <vkaraf@gmail.com>
Sat, 8 Sep 2018 10:31:11 +0000 (12:31 +0200)
committerShanigen <vkaraf@gmail.com>
Sat, 8 Sep 2018 10:31:11 +0000 (12:31 +0200)
These problems were found by Travis CL.

src/complexmat.cuh
src/kcf.cpp

index a9d76cd480faf6b1246230a10a43329b44f7505b..57ccee3f937dc20b71f46a60c0570b76a55139b5 100644 (file)
@@ -106,14 +106,14 @@ class ComplexMat {
     // text output
     friend std::ostream &operator<<(std::ostream &os, const ComplexMat &mat)
     {
-        float *data_cpu = (float *)malloc(mat.rows * mat.cols * mat.n_channels * sizeof(cufftComplex));
+        float *data_cpu = reinterpret_cast<float*>(malloc(mat.rows * mat.cols * mat.n_channels * sizeof(cufftComplex)));
         CudaSafeCall(cudaMemcpy(data_cpu, mat.p_data, mat.rows * mat.cols * mat.n_channels * sizeof(cufftComplex),
                                 cudaMemcpyDeviceToHost));
         // for (int i = 0; i < mat.n_channels; ++i){
         for (int i = 0; i < 1; ++i) {
             os << "Channel " << i << std::endl;
-            for (int j = 0; j < mat.rows; ++j) {
-                for (int k = 0; k < 2 * mat.cols - 2; k += 2)
+            for (uint j = 0; j < mat.rows; ++j) {
+                for (uint k = 0; k < 2 * mat.cols - 2; k += 2)
                     os << "(" << data_cpu[j * 2 * mat.cols + k] << "," << data_cpu[j * 2 * mat.cols + (k + 1)] << ")"
                        << ", ";
                 os << "(" << data_cpu[j * 2 * mat.cols + 2 * mat.cols - 2] << ","
index 9c6554bb513c481f9bed4cdd0d01d7e8cd4b6726..470c4485e11729a26f1abf196f137ae553aefde4 100644 (file)
@@ -155,10 +155,9 @@ void KCF_Tracker::init(cv::Mat &img, const cv::Rect &bbox, int fit_size_x, int f
         std::exit(EXIT_FAILURE);
     }
     CudaSafeCall(cudaSetDeviceFlags(cudaDeviceMapHost));
-    p_xf.create(uint(p_windows_size.height) / p_cell_size, (uint(p_windows_size.width) / p_cell_size) / 2 + 1, p_num_of_feats, this->stream);
     p_rot_labels_data = DynMem(
         ((uint(p_windows_size.width) / p_cell_size) * (uint(p_windows_size.height) / p_cell_size)) * sizeof(float));
-    p_rot_labels = cv::Mat(p_windows_size.height / int(p_cell_size), windows_size.width / int(p_cell_size), CV_32FC1,
+    p_rot_labels = cv::Mat(p_windows_size.height / int(p_cell_size), p_windows_size.width / int(p_cell_size), CV_32FC1,
                            p_rot_labels_data.hostMem());
 #else
     p_xf.create(uint(p_windows_size.height / p_cell_size), (uint(p_windows_size.height / p_cell_size)) / 2 + 1,