From 0cba003cb8c912c36fbb6672c4355dc0ba867925 Mon Sep 17 00:00:00 2001
From: xizero00 <xizero00@gmail.com>
Date: Wed, 26 Jul 2017 15:49:43 +0800
Subject: [PATCH] modify Makefile to link with opencv_contrib and change
 blocking_queue.cpp to add two headers and Blockqueue with DataReader and
 Datum

---
 Makefile                          | 2 +-
 src/caffe/util/blocking_queue.cpp | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/Makefile b/Makefile
index ad173f80..b50c5581 100755
--- a/Makefile
+++ b/Makefile
@@ -192,7 +192,7 @@ ifeq ($(USE_LMDB), 1)
 	LIBRARIES += lmdb
 endif
 ifeq ($(USE_OPENCV), 1)
-	LIBRARIES += opencv_core opencv_highgui opencv_imgproc
+	LIBRARIES += opencv_core opencv_highgui opencv_imgproc opencv_contrib
 
 	ifeq ($(OPENCV_VERSION), 3)
 		LIBRARIES += opencv_imgcodecs
diff --git a/src/caffe/util/blocking_queue.cpp b/src/caffe/util/blocking_queue.cpp
index f69d2104..e8b89e1b 100755
--- a/src/caffe/util/blocking_queue.cpp
+++ b/src/caffe/util/blocking_queue.cpp
@@ -2,6 +2,8 @@
 #include <string>
 
 #include "caffe/layers/base_data_layer.hpp"
+#include "caffe/data_reader.hpp"
+#include "caffe/proto/caffe.pb.h"
 #include "caffe/parallel.hpp"
 #include "caffe/util/blocking_queue.hpp"
 
@@ -87,5 +89,7 @@ size_t BlockingQueue<T>::size() const {
 
 template class BlockingQueue<Batch<float>*>;
 template class BlockingQueue<Batch<double>*>;
+template class BlockingQueue<Datum*>;
+template class BlockingQueue<shared_ptr<DataReader::QueuePair> >;
 
 }  // namespace caffe