From: Sergio <sguada@gmail.com>
Date: Fri, 26 Sep 2014 02:53:33 +0000 (-0700)
Subject: Fix OSX compilation for nvcc with opencv
X-Git-Tag: submit/tizen/20180823.020014~572^2~127^2~4
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=8cd863e7cb5181cbbdfcf0dd75a78c80c56c1281;p=platform%2Fupstream%2Fcaffeonacl.git

Fix OSX compilation for nvcc with opencv
---

diff --git a/Makefile b/Makefile
index b0b2515..7dea267 100644
--- a/Makefile
+++ b/Makefile
@@ -252,6 +252,7 @@ ifeq ($(OSX), 1)
 	endif
 	# boost::thread is called boost_thread-mt to mark multithreading on OS X
 	LIBRARIES += boost_thread-mt
+        NVCCFLAGS += -DOSX
 endif
 
 # Custom compiler
diff --git a/include/caffe/data_transformer.hpp b/include/caffe/data_transformer.hpp
index 735f1d5..7e48df3 100644
--- a/include/caffe/data_transformer.hpp
+++ b/include/caffe/data_transformer.hpp
@@ -1,7 +1,9 @@
 #ifndef CAFFE_DATA_TRANSFORMER_HPP
 #define CAFFE_DATA_TRANSFORMER_HPP
 
+#ifndef OSX
 #include <opencv2/core/core.hpp>
+#endif
 
 #include <vector>
 
@@ -49,7 +51,9 @@ class DataTransformer {
   void Transform(const vector<Datum> & datum_vector,
                 Blob<Dtype>* transformed_blob);
 
+#ifndef OSX
   void Transform(const cv::Mat& cv_img, Blob<Dtype>* transformed_blob);
+#endif
 
   void Transform(Blob<Dtype>* input_blob, Blob<Dtype>* transformed_blob);
 
diff --git a/include/caffe/util/io.hpp b/include/caffe/util/io.hpp
index 2032b1f..e518979 100644
--- a/include/caffe/util/io.hpp
+++ b/include/caffe/util/io.hpp
@@ -1,7 +1,9 @@
 #ifndef CAFFE_UTIL_IO_H_
 #define CAFFE_UTIL_IO_H_
 
+#ifndef OSX
 #include <opencv2/core/core.hpp>
+#endif
 
 #include <unistd.h>
 #include <string>
@@ -109,6 +111,7 @@ inline bool ReadImageToDatum(const string& filename, const int label,
   return ReadImageToDatum(filename, label, 0, 0, true, datum);
 }
 
+#ifndef OSX
 cv::Mat ReadImageToCVMat(const string& filename,
     const int height, const int width, const bool is_color);
 
@@ -127,6 +130,7 @@ inline cv::Mat ReadImageToCVMat(const string& filename) {
 }
 
 void CVMatToDatum(const cv::Mat& cv_img, Datum* datum);
+#endif
 
 leveldb::Options GetLevelDBOptions();
 
diff --git a/models/bvlc_reference_caffenet/train_val.prototxt b/models/bvlc_reference_caffenet/train_val.prototxt
index 073d8ae..6cfeab1 100644
--- a/models/bvlc_reference_caffenet/train_val.prototxt
+++ b/models/bvlc_reference_caffenet/train_val.prototxt
@@ -330,11 +330,20 @@ layers {
   }
 }
 layers {
-  name: "accuracy"
+  name: "top-1"
   type: ACCURACY
   bottom: "fc8"
   bottom: "label"
-  top: "accuracy"
+  top: "top-1"
+  include: { phase: TEST }
+}
+layers {
+  name: "top-5"
+  type: ACCURACY
+  bottom: "fc8"
+  bottom: "label"
+  top: "top-5"
+  accuracy_param: { top_k: 5 }
   include: { phase: TEST }
 }
 layers {
diff --git a/src/caffe/data_transformer.cpp b/src/caffe/data_transformer.cpp
index abc6ec2..a195c5d 100644
--- a/src/caffe/data_transformer.cpp
+++ b/src/caffe/data_transformer.cpp
@@ -1,4 +1,6 @@
+#ifndef OSX
 #include <opencv2/core/core.hpp>
+#endif
 
 #include <string>
 #include <vector>
@@ -172,6 +174,7 @@ void DataTransformer<Dtype>::Transform(const vector<Datum> & datum_vector,
   }
 }
 
+#ifndef OSX
 template<typename Dtype>
 void DataTransformer<Dtype>::Transform(const cv::Mat& cv_img,
                                        Blob<Dtype>* transformed_blob) {
@@ -179,6 +182,7 @@ void DataTransformer<Dtype>::Transform(const cv::Mat& cv_img,
   CVMatToDatum(cv_img, &datum);
   Transform(datum, transformed_blob);
 }
+#endif
 
 template<typename Dtype>
 void DataTransformer<Dtype>::Transform(Blob<Dtype>* input_blob,