[DetectNetTransform] Build with GPU and w/o OpenCV

wangjuenew · Oct 12, 2016 · 7af32cd · 7af32cd
1 parent fe72427
commit 7af32cd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/caffe/layers/detectnet_transform_layer.cu b/src/caffe/layers/detectnet_transform_layer.cu
@@ -1,3 +1,5 @@
+#ifdef USE_OPENCV
+
 #include <math.h>
 #include <math_constants.h>
 #include <opencv2/core/core.hpp>
@@ -430,3 +432,5 @@ INSTANTIATE_LAYER_GPU_FUNCS(DetectNetTransformationLayer);
 
 
 }  // namespace caffe
+
+#endif