fixded PCA issues

Colin Heinzmann
1 parent b17fcc31
Showing 2 changed files with 7 additions and 0 deletions
openbr/plugins/cuda/cudapca.cpp
openbr/plugins/cuda/cudapca.cu
@@ -33,6 +33,7 @@ using namespace cv;
 // definitions from the CUDA source file
 namespace br { namespace cuda { namespace pca {
   void initializeWrapper(float* evPtr, int evRows, int evCols, float* meanPtr, int meanElems);
+  void trainWrapper(void* cudaSrc, float* dst, int rows, int cols);
   void wrapper(void* src, void** dst);
 }}}
@@ -94,6 +95,7 @@ private:
           int type = *((int*)srcDataPtr[3]);
           Mat mat = Mat(rows, cols, type);
+          br::cuda::pca::trainWrapper(cudaMemPtr, mat.ptr<float>(), rows, cols);
           trainingQlist.append(Template(mat));
         }
@@ -111,6 +111,11 @@ namespace br { namespace cuda { namespace pca {
     CUDA_SAFE_MALLOC(&intermediaryBuffer, _numSteps*_evCols*sizeof(float), &err);
   }
+  void trainWrapper(void* cudaSrc, float* data, int rows, int cols) {
+    cudaError_t err;
+    CUDA_SAFE_MEMCPY(data, cudaSrc, rows*cols*sizeof(float), cudaMemcpyDeviceToHost, &err);
+  }
+
   void wrapper(void* src, void** dst) {
     cudaError_t err;
     CUDA_SAFE_MALLOC(dst, _evCols*sizeof(float), &err);