jackcc · jackcc · Jun 6, 2016 · Jun 6, 2016 · Jun 7, 2016 · Jun 8, 2016
diff --git a/.github/ISSUE_TEMPLATE.md b/.github/ISSUE_TEMPLATE.md
@@ -0,0 +1,19 @@
+Please use the [caffe-users list](https://groups.google.com/forum/#!forum/caffe-users) for usage, installation, or modeling questions, or other requests for help.
+_Do not post such requests to Issues._ Doing so interferes with the development of Caffe.
+
+Please read the [guidelines for contributing](https://github.com/BVLC/caffe/blob/master/CONTRIBUTING.md) before submitting this issue.
+
+### Issue summary
+
+
+### Steps to reproduce
+
+If you are having difficulty building Caffe or training a model, please ask the caffe-users mailing list. If you are reporting a build error that seems to be due to a bug in Caffe, please attach your build configuration (either Makefile.config or CMakeCache.txt) and the output of the make (or cmake) command.
+
+### Your system configuration
+Operating system:
+Compiler:
+CUDA version (if applicable):
+CUDNN version (if applicable):
+BLAS:
+Python or MATLAB version (for pycaffe and matcaffe respectively):
diff --git a/.gitignore b/.gitignore
@@ -47,6 +47,9 @@
 # PyCharm files
 .idea
 
+# Visual Studio Code files
+.vscode
+
 # OSX dir files
 .DS_Store
 
@@ -93,3 +96,8 @@ LOCK
 LOG*
 CURRENT
 MANIFEST-*
+
+# temporary directories
+jobs
+temp
+examples/*/*lmdb
diff --git a/.travis.yml b/.travis.yml
@@ -28,7 +28,6 @@ env:
     - BUILD_NAME="cudnn-cmake" WITH_CMAKE=true WITH_CUDA=true WITH_CUDNN=true
 
 cache:
-  timeout: 604800  # 1 week
   apt: true
   directories:
     - ~/protobuf3

diff --git a/LICENSE b/LICENSE
@@ -1,5 +1,11 @@
 COPYRIGHT
 
+All new contributions compared to the original branch:
+Copyright (c) 2015, 2016 Wei Liu (UNC Chapel Hill), Dragomir Anguelov (Zoox),
+Dumitru Erhan (Google), Christian Szegedy (Google), Scott Reed (UMich Ann Arbor),
+Cheng-Yang Fu (UNC Chapel Hill), Alexander C. Berg (UNC Chapel Hill).
+All rights reserved.
+
 All contributions by the University of California:
 Copyright (c) 2014, 2015, The Regents of the University of California (Regents)
 All rights reserved.

diff --git a/Makefile b/Makefile
@@ -178,7 +178,7 @@ ifneq ($(CPU_ONLY), 1)
 	LIBRARIES := cudart cublas curand
 endif
 
-LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
+LIBRARIES += glog gflags protobuf boost_system boost_filesystem boost_regex m hdf5_hl hdf5
 
 # handle IO dependencies
 USE_LEVELDB ?= 1
@@ -192,12 +192,12 @@ 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
 
 	ifeq ($(OPENCV_VERSION), 3)
-		LIBRARIES += opencv_imgcodecs
+		LIBRARIES += opencv_imgcodecs opencv_videoio
 	endif
-		
+
 endif
 PYTHON_LIBRARIES ?= boost_python python2.7
 WARNINGS := -Wall -Wno-sign-compare
@@ -382,8 +382,11 @@ else
 		LIBRARIES += cblas
 		# 10.10 has accelerate while 10.9 has veclib
 		XCODE_CLT_VER := $(shell pkgutil --pkg-info=com.apple.pkg.CLTools_Executables | grep 'version' | sed 's/[^0-9]*\([0-9]\).*/\1/')
+		XCODE_CLT_GEQ_7 := $(shell [ $(XCODE_CLT_VER) -gt 6 ] && echo 1)
 		XCODE_CLT_GEQ_6 := $(shell [ $(XCODE_CLT_VER) -gt 5 ] && echo 1)
-		ifeq ($(XCODE_CLT_GEQ_6), 1)
+		ifeq ($(XCODE_CLT_GEQ_7), 1)
+			BLAS_INCLUDE ?= /Applications/Xcode.app/Contents/Developer/Platforms/MacOSX.platform/Developer/SDKs/$(shell ls /Applications/Xcode.app/Contents/Developer/Platforms/MacOSX.platform/Developer/SDKs/ | sort | tail -1)/System/Library/Frameworks/Accelerate.framework/Versions/A/Frameworks/vecLib.framework/Versions/A/Headers
+		else ifeq ($(XCODE_CLT_GEQ_6), 1)
 			BLAS_INCLUDE ?= /System/Library/Frameworks/Accelerate.framework/Versions/Current/Frameworks/vecLib.framework/Headers/
 			LDFLAGS += -framework Accelerate
 		else
@@ -401,7 +404,7 @@ LIBRARY_DIRS += $(LIB_BUILD_DIR)
 CXXFLAGS += -MMD -MP
 
 # Complete build flags.
-COMMON_FLAGS += $(foreach includedir,$(INCLUDE_DIRS),-I$(includedir))
+COMMON_FLAGS += $(foreach includedir,$(INCLUDE_DIRS),-isystem $(includedir))
 CXXFLAGS += -pthread -fPIC $(COMMON_FLAGS) $(WARNINGS)
 NVCCFLAGS += -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
 # mex may invoke an older gcc that is too liberal with -Wuninitalized

diff --git a/Makefile.config.example b/Makefile.config.example
@@ -31,19 +31,21 @@ CUDA_DIR := /usr/local/cuda
 # CUDA_DIR := /usr
 
 # CUDA architecture setting: going with all of them.
-# For CUDA < 6.0, comment the *_50 lines for compatibility.
+# For CUDA < 6.0, comment the lines after *_35 for compatibility.
 CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
-		-gencode arch=compute_20,code=sm_21 \
-		-gencode arch=compute_30,code=sm_30 \
-		-gencode arch=compute_35,code=sm_35 \
-		-gencode arch=compute_50,code=sm_50 \
-		-gencode arch=compute_50,code=compute_50
+             -gencode arch=compute_20,code=sm_21 \
+             -gencode arch=compute_30,code=sm_30 \
+             -gencode arch=compute_35,code=sm_35 \
+             -gencode arch=compute_50,code=sm_50 \
+             -gencode arch=compute_52,code=sm_52 \
+             -gencode arch=compute_61,code=sm_61
 
 # BLAS choice:
 # atlas for ATLAS (default)
 # mkl for MKL
 # open for OpenBlas
-BLAS := atlas
+# BLAS := atlas
+BLAS := open
 # Custom (MKL/ATLAS/OpenBLAS) include and lib directories.
 # Leave commented to accept the defaults for your choice of BLAS
 # (which should work)!
@@ -65,10 +67,10 @@ PYTHON_INCLUDE := /usr/include/python2.7 \
 		/usr/lib/python2.7/dist-packages/numpy/core/include
 # Anaconda Python distribution is quite popular. Include path:
 # Verify anaconda location, sometimes it's in root.
-# ANACONDA_HOME := $(HOME)/anaconda
+# ANACONDA_HOME := $(HOME)/anaconda2
 # PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
-		# $(ANACONDA_HOME)/include/python2.7 \
-		# $(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include \
+		$(ANACONDA_HOME)/include/python2.7 \
+		$(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include \
 
 # Uncomment to use Python 3 (default is Python 2)
 # PYTHON_LIBRARIES := boost_python3 python3.5m

diff --git a/README.md b/README.md
@@ -1,37 +1,143 @@
-# Caffe
+# SSD: Single Shot MultiBox Detector
 
-[![Build Status](https://travis-ci.org/BVLC/caffe.svg?branch=master)](https://travis-ci.org/BVLC/caffe)
+[![Build Status](https://travis-ci.org/weiliu89/caffe.svg?branch=ssd)](https://travis-ci.org/weiliu89/caffe)
 [![License](https://img.shields.io/badge/license-BSD-blue.svg)](LICENSE)
 
-Caffe is a deep learning framework made with expression, speed, and modularity in mind.
-It is developed by the Berkeley Vision and Learning Center ([BVLC](http://bvlc.eecs.berkeley.edu)) and community contributors.
+By [Wei Liu](http://www.cs.unc.edu/~wliu/), [Dragomir Anguelov](https://www.linkedin.com/in/dragomiranguelov), [Dumitru Erhan](http://research.google.com/pubs/DumitruErhan.html), [Christian Szegedy](http://research.google.com/pubs/ChristianSzegedy.html), [Scott Reed](http://www-personal.umich.edu/~reedscot/), [Cheng-Yang Fu](http://www.cs.unc.edu/~cyfu/), [Alexander C. Berg](http://acberg.com).
 
-Check out the [project site](http://caffe.berkeleyvision.org) for all the details like
+### Introduction
 
-- [DIY Deep Learning for Vision with Caffe](https://docs.google.com/presentation/d/1UeKXVgRvvxg9OUdh_UiC5G71UMscNPlvArsWER41PsU/edit#slide=id.p)
-- [Tutorial Documentation](http://caffe.berkeleyvision.org/tutorial/)
-- [BVLC reference models](http://caffe.berkeleyvision.org/model_zoo.html) and the [community model zoo](https://github.com/BVLC/caffe/wiki/Model-Zoo)
-- [Installation instructions](http://caffe.berkeleyvision.org/installation.html)
+SSD is an unified framework for object detection with a single network. You can use the code to train/evaluate a network for object detection task. For more details, please refer to our [arXiv paper](http://arxiv.org/abs/1512.02325) and our [slide](http://www.cs.unc.edu/~wliu/papers/ssd_eccv2016_slide.pdf).
 
-and step-by-step examples.
+<p align="center">
+<img src="http://www.cs.unc.edu/~wliu/papers/ssd.png" alt="SSD Framework" width="600px">
+</p>
 
-[![Join the chat at https://gitter.im/BVLC/caffe](https://badges.gitter.im/Join%20Chat.svg)](https://gitter.im/BVLC/caffe?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge)
+| System | VOC2007 test *mAP* | **FPS** (Titan X) | Number of Boxes | Input resolution
+|:-------|:-----:|:-------:|:-------:|:-------:|
+| [Faster R-CNN (VGG16)](https://github.com/ShaoqingRen/faster_rcnn) | 73.2 | 7 | ~6000 | ~1000 x 600 |
+| [YOLO (customized)](http://pjreddie.com/darknet/yolo/) | 63.4 | 45 | 98 | 448 x 448 |
+| SSD300* (VGG16) | 77.2 | 46 | 8732 | 300 x 300 |
+| SSD512* (VGG16) | **79.8** | 19 | 24564 | 512 x 512 |
 
-Please join the [caffe-users group](https://groups.google.com/forum/#!forum/caffe-users) or [gitter chat](https://gitter.im/BVLC/caffe) to ask questions and talk about methods and models.
-Framework development discussions and thorough bug reports are collected on [Issues](https://github.com/BVLC/caffe/issues).
 
-Happy brewing!
+<p align="left">
+<img src="http://www.cs.unc.edu/~wliu/papers/ssd_results.png" alt="SSD results on multiple datasets" width="800px">
+</p>
 
-## License and Citation
+_Note: SSD300* and SSD512* are the latest models. Current code should reproduce these results._
 
-Caffe is released under the [BSD 2-Clause license](https://github.com/BVLC/caffe/blob/master/LICENSE).
-The BVLC reference models are released for unrestricted use.
+### Citing SSD
 
-Please cite Caffe in your publications if it helps your research:
+Please cite SSD in your publications if it helps your research:
 
-    @article{jia2014caffe,
-      Author = {Jia, Yangqing and Shelhamer, Evan and Donahue, Jeff and Karayev, Sergey and Long, Jonathan and Girshick, Ross and Guadarrama, Sergio and Darrell, Trevor},
-      Journal = {arXiv preprint arXiv:1408.5093},
-      Title = {Caffe: Convolutional Architecture for Fast Feature Embedding},
-      Year = {2014}
+    @inproceedings{liu2016ssd,
+      title = {{SSD}: Single Shot MultiBox Detector},
+      author = {Liu, Wei and Anguelov, Dragomir and Erhan, Dumitru and Szegedy, Christian and Reed, Scott and Fu, Cheng-Yang and Berg, Alexander C.},
+      booktitle = {ECCV},
+      year = {2016}
     }
+
+### Contents
+1. [Installation](#installation)
+2. [Preparation](#preparation)
+3. [Train/Eval](#traineval)
+4. [Models](#models)
+
+### Installation
+1. Get the code. We will call the directory that you cloned Caffe into `$CAFFE_ROOT`
+  ```Shell
+  git clone https://github.com/weiliu89/caffe.git
+  cd caffe
+  git checkout ssd
+  ```
+
+2. Build the code. Please follow [Caffe instruction](http://caffe.berkeleyvision.org/installation.html) to install all necessary packages and build it.
+  ```Shell
+  # Modify Makefile.config according to your Caffe installation.
+  cp Makefile.config.example Makefile.config
+  make -j8
+  # Make sure to include $CAFFE_ROOT/python to your PYTHONPATH.
+  make py
+  make test -j8
+  # (Optional)
+  make runtest -j8
+  ```
+
+### Preparation
+1. Download [fully convolutional reduced (atrous) VGGNet](https://gist.github.com/weiliu89/2ed6e13bfd5b57cf81d6). By default, we assume the model is stored in `$CAFFE_ROOT/models/VGGNet/`
+
+2. Download VOC2007 and VOC2012 dataset. By default, we assume the data is stored in `$HOME/data/`
+  ```Shell
+  # Download the data.
+  cd $HOME/data
+  wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
+  wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
+  wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
+  # Extract the data.
+  tar -xvf VOCtrainval_11-May-2012.tar
+  tar -xvf VOCtrainval_06-Nov-2007.tar
+  tar -xvf VOCtest_06-Nov-2007.tar
+  ```
+
+3. Create the LMDB file.
+  ```Shell
+  cd $CAFFE_ROOT
+  # Create the trainval.txt, test.txt, and test_name_size.txt in data/VOC0712/
+  ./data/VOC0712/create_list.sh
+  # You can modify the parameters in create_data.sh if needed.
+  # It will create lmdb files for trainval and test with encoded original image:
+  #   - $HOME/data/VOCdevkit/VOC0712/lmdb/VOC0712_trainval_lmdb
+  #   - $HOME/data/VOCdevkit/VOC0712/lmdb/VOC0712_test_lmdb
+  # and make soft links at examples/VOC0712/
+  ./data/VOC0712/create_data.sh
+  ```
+
+### Train/Eval
+1. Train your model and evaluate the model on the fly.
+  ```Shell
+  # It will create model definition files and save snapshot models in:
+  #   - $CAFFE_ROOT/models/VGGNet/VOC0712/SSD_300x300/
+  # and job file, log file, and the python script in:
+  #   - $CAFFE_ROOT/jobs/VGGNet/VOC0712/SSD_300x300/
+  # and save temporary evaluation results in:
+  #   - $HOME/data/VOCdevkit/results/VOC2007/SSD_300x300/
+  # It should reach 77.* mAP at 120k iterations.
+  python examples/ssd/ssd_pascal.py
+  ```
+  If you don't have time to train your model, you can download a pre-trained model at [here](https://drive.google.com/file/d/0BzKzrI_SkD1_WVVTSmQxU0dVRzA/view?resourcekey=0-VFoYwzlHsh486pZ5U4sA6w).
+
+2. Evaluate the most recent snapshot.
+  ```Shell
+  # If you would like to test a model you trained, you can do:
+  python examples/ssd/score_ssd_pascal.py
+  ```
+
+3. Test your model using a webcam. Note: press <kbd>esc</kbd> to stop.
+  ```Shell
+  # If you would like to attach a webcam to a model you trained, you can do:
+  python examples/ssd/ssd_pascal_webcam.py
+  ```
+  [Here](https://drive.google.com/file/d/0BzKzrI_SkD1_R09NcjM1eElLcWc/view?resourcekey=0-Sx3DJK4udeqqVl7AZkfLlQ) is a demo video of running a SSD500 model trained on [MSCOCO](http://mscoco.org) dataset.
+
+4. Check out [`examples/ssd_detect.ipynb`](https://github.com/weiliu89/caffe/blob/ssd/examples/ssd_detect.ipynb) or [`examples/ssd/ssd_detect.cpp`](https://github.com/weiliu89/caffe/blob/ssd/examples/ssd/ssd_detect.cpp) on how to detect objects using a SSD model. Check out [`examples/ssd/plot_detections.py`](https://github.com/weiliu89/caffe/blob/ssd/examples/ssd/plot_detections.py) on how to plot detection results output by ssd_detect.cpp.
+
+5. To train on other dataset, please refer to data/OTHERDATASET for more details. We currently add support for COCO and ILSVRC2016. We recommend using [`examples/ssd.ipynb`](https://github.com/weiliu89/caffe/blob/ssd/examples/ssd_detect.ipynb) to check whether the new dataset is prepared correctly.
+
+### Models
+We have provided the latest models that are trained from different datasets. To help reproduce the results in [Table 6](https://arxiv.org/pdf/1512.02325v4.pdf), most models contain a pretrained `.caffemodel` file, many `.prototxt` files, and python scripts.
+
+1. PASCAL VOC models:
+   * 07+12: [SSD300*](https://drive.google.com/file/d/0BzKzrI_SkD1_WVVTSmQxU0dVRzA/view?usp=sharing&resourcekey=0-VFoYwzlHsh486pZ5U4sA6w), [SSD512*](https://drive.google.com/file/d/0BzKzrI_SkD1_ZDIxVHBEcUNBb2s/view?resourcekey=0-cK6N_2Rq6C5wHAo8RaKHFw)
+   * 07++12: [SSD300*](https://drive.google.com/file/d/0BzKzrI_SkD1_WnR2T1BGVWlCZHM/view?resourcekey=0-JuQ-1PI4gfEU5Y9hY6IXbg), [SSD512*](https://drive.google.com/file/d/0BzKzrI_SkD1_MjFjNTlnempHNWs/view?resourcekey=0-0K7EITSbE2Fae2pIMUrpSQ)
+   * COCO<sup>[1]</sup>: [SSD300*](https://drive.google.com/file/d/0BzKzrI_SkD1_NDlVeFJDc2tIU1k/view?resourcekey=0-VIwceFdQvGpMl31jHv5RpA), [SSD512*](https://drive.google.com/file/d/0BzKzrI_SkD1_TW4wTC14aDdCTDQ/view?resourcekey=0-KGdQFztFWQmhnOcIH6rHRg)
+   * 07+12+COCO: [SSD300*](https://drive.google.com/file/d/0BzKzrI_SkD1_UFpoU01yLS1SaG8/view?resourcekey=0-WTJQYLuiLBUlNJl8R8qLBQ), [SSD512*](https://drive.google.com/file/d/0BzKzrI_SkD1_X3ZXQUUtM0xNeEk/view?resourcekey=0-lE4MISx92j4QfcQB0IG8tQ)
+   * 07++12+COCO: [SSD300*](https://drive.google.com/file/d/0BzKzrI_SkD1_TkFPTEQ1Z091SUE/view?resourcekey=0-y2qRDQXEnu7vfOOVtOv4zQ), [SSD512*](https://drive.google.com/file/d/0BzKzrI_SkD1_NVVNdWdYNEh1WTA/view?resourcekey=0-w1KTy5Kf6iSXHc0Njd_btg)
+
+2. COCO models:
+   * trainval35k: [SSD300*](https://drive.google.com/file/d/0BzKzrI_SkD1_dUY1Ml9GRTFpUWc/view?resourcekey=0-fJphZuOKnVT1Pkd89oXSlg), [SSD512*](https://drive.google.com/file/d/0BzKzrI_SkD1_dlJpZHJzOXd3MTg/view?resourcekey=0-RmOcjeeLhvcJYlQqLf-LJw)
+
+3. ILSVRC models:
+   * trainval1: [SSD300*](https://drive.google.com/file/d/0BzKzrI_SkD1_a2NKQ2d1d043VXM/view?resourcekey=0-qLYN1paAkTpJYc91FOAqKw), [SSD500](https://drive.google.com/file/d/0BzKzrI_SkD1_X2ZCLVgwLTgzaTQ/view?resourcekey=0-GA44ga07kV8oB0rK7XEF0Q)
+
+<sup>[1]</sup>We use [`examples/convert_model.ipynb`](https://github.com/weiliu89/caffe/blob/ssd/examples/convert_model.ipynb) to extract a VOC model from a pretrained COCO model.
diff --git a/cmake/Cuda.cmake b/cmake/Cuda.cmake
@@ -4,7 +4,7 @@ endif()
 
 # Known NVIDIA GPU achitectures Caffe can be compiled for.
 # This list will be used for CUDA_ARCH_NAME = All option
-set(Caffe_known_gpu_archs "20 21(20) 30 35 50")
+set(Caffe_known_gpu_archs "20 21(20) 30 35 50 52 61")
 
 ################################################################################################
 # A function for automatic detection of GPUs installed  (if autodetection is enabled)
@@ -174,11 +174,18 @@ function(detect_cuDNN)
             PATHS ${CUDNN_ROOT} $ENV{CUDNN_ROOT} ${CUDA_TOOLKIT_INCLUDE}
             DOC "Path to cuDNN include directory." )
 
-  get_filename_component(__libpath_hist ${CUDA_CUDART_LIBRARY} PATH)
-  find_library(CUDNN_LIBRARY NAMES libcudnn.so # libcudnn_static.a
-                             PATHS ${CUDNN_ROOT} $ENV{CUDNN_ROOT} ${CUDNN_INCLUDE} ${__libpath_hist}
-                             DOC "Path to cuDNN library.")
+  # dynamic libs have different suffix in mac and linux
+  if(APPLE)
+    set(CUDNN_LIB_NAME "libcudnn.dylib")
+  else()
+    set(CUDNN_LIB_NAME "libcudnn.so")
+  endif()
 
+  get_filename_component(__libpath_hist ${CUDA_CUDART_LIBRARY} PATH)
+  find_library(CUDNN_LIBRARY NAMES ${CUDNN_LIB_NAME}
+   PATHS ${CUDNN_ROOT} $ENV{CUDNN_ROOT} ${CUDNN_INCLUDE} ${__libpath_hist} ${__libpath_hist}/../lib
+   DOC "Path to cuDNN library.")
+
   if(CUDNN_INCLUDE AND CUDNN_LIBRARY)
     set(HAVE_CUDNN  TRUE PARENT_SCOPE)
     set(CUDNN_FOUND TRUE PARENT_SCOPE)

diff --git a/cmake/Dependencies.cmake b/cmake/Dependencies.cmake
@@ -2,7 +2,7 @@
 set(Caffe_LINKER_LIBS "")
 
 # ---[ Boost
-find_package(Boost 1.46 REQUIRED COMPONENTS system thread filesystem)
+find_package(Boost 1.46 REQUIRED COMPONENTS system thread filesystem regex)
 include_directories(SYSTEM ${Boost_INCLUDE_DIR})
 list(APPEND Caffe_LINKER_LIBS ${Boost_LIBRARIES})
 
@@ -26,7 +26,7 @@ include(cmake/ProtoBuf.cmake)
 # ---[ HDF5
 find_package(HDF5 COMPONENTS HL REQUIRED)
 include_directories(SYSTEM ${HDF5_INCLUDE_DIRS} ${HDF5_HL_INCLUDE_DIR})
-list(APPEND Caffe_LINKER_LIBS ${HDF5_LIBRARIES})
+list(APPEND Caffe_LINKER_LIBS ${HDF5_LIBRARIES} ${HDF5_HL_LIBRARIES})
 
 # ---[ LMDB
 if(USE_LMDB)
@@ -69,7 +69,7 @@ endif()
 
 # ---[ OpenCV
 if(USE_OPENCV)
-  find_package(OpenCV QUIET COMPONENTS core highgui imgproc imgcodecs)
+  find_package(OpenCV QUIET COMPONENTS core highgui imgproc imgcodecs videoio)
   if(NOT OpenCV_FOUND) # if not OpenCV 3.x, then imgcodecs are not found
     find_package(OpenCV REQUIRED COMPONENTS core highgui imgproc)
   endif()
@@ -102,6 +102,12 @@ elseif(APPLE)
   find_package(vecLib REQUIRED)
   include_directories(SYSTEM ${vecLib_INCLUDE_DIR})
   list(APPEND Caffe_LINKER_LIBS ${vecLib_LINKER_LIBS})
+
+  if(VECLIB_FOUND)
+    if(NOT vecLib_INCLUDE_DIR MATCHES "^/System/Library/Frameworks/vecLib.framework.*")
+      add_definitions(-DUSE_ACCELERATE)
+    endif()
+  endif()
 endif()
 
 # ---[ Python