diff --git a/include/caffe/vision_layers.hpp b/include/caffe/vision_layers.hpp index 74b27ccf..a511f922 100644 --- a/include/caffe/vision_layers.hpp +++ b/include/caffe/vision_layers.hpp @@ -12,6 +12,8 @@ #include "boost/scoped_ptr.hpp" #include "hdf5.h" +#include "caffe/blob.hpp" +#include "caffe/common.hpp" #include "caffe/layer.hpp" #include "caffe/proto/caffe.pb.h" diff --git a/src/caffe/layers/images_layer.cu b/src/caffe/layers/images_layer.cu index 7f67283e..187be33a 100644 --- a/src/caffe/layers/images_layer.cu +++ b/src/caffe/layers/images_layer.cu @@ -1,5 +1,6 @@ // Copyright 2013 Yangqing Jia +#include #include #include #include @@ -9,6 +10,8 @@ #include // NOLINT(readability/streams) #include // NOLINT(readability/streams) +#include "caffe/blob.hpp" +#include "caffe/common.hpp" #include "caffe/layer.hpp" #include "caffe/util/io.hpp" #include "caffe/vision_layers.hpp" diff --git a/src/caffe/layers/window_data_layer.cu b/src/caffe/layers/window_data_layer.cu index bf331941..3994352e 100644 --- a/src/caffe/layers/window_data_layer.cu +++ b/src/caffe/layers/window_data_layer.cu @@ -2,6 +2,7 @@ // // Based on data_layer.cpp by Yangqing Jia. +#include #include #include @@ -16,6 +17,8 @@ #include "opencv2/highgui/highgui.hpp" #include "opencv2/imgproc/imgproc.hpp" +#include "caffe/blob.hpp" +#include "caffe/common.hpp" #include "caffe/layer.hpp" #include "caffe/util/io.hpp" #include "caffe/vision_layers.hpp"