Detection

.gitignore

@@ -26,21 +26,29 @@
 # Compiled MATLAB
 *.mex*
 
-# build, distribute, and bins
-build
-.build_debug/*
-.build_release/*
-distribute/*
-*.testbin
-*.bin
-python/caffe/proto/
+# IPython notebook checkpoints
+.ipynb_checkpoints
 
 # Editor temporaries
 *.swp
 *~
 
-# IPython notebook checkpoints
-.ipynb_checkpoints
+# Sublime Text settings
+*.sublime-workspace
+*.sublime-project
+
+# Eclipse Project settings
+*.*project
+.settings
+
+# QtCreator files
+*.user
+
+# PyCharm files
+.idea
+
+# OSX dir files
+.DS_Store
 
 ## Caffe
 
@@ -58,13 +66,16 @@ models/*
 *leveldb
 *lmdb
 
-# LevelDB files
-*.sst
-*.ldb
-LOCK
-LOG*
-CURRENT
-MANIFEST-*
+# build, distribute, and bins (+ python proto bindings)
+build
+.build_debug/*
+.build_release/*
+distribute/*
+*.testbin
+*.bin
+python/caffe/proto/
+cmake_build
+.cmake_build
 
 # Generated documentation
 docs/_site
@@ -73,12 +84,10 @@ _site
 doxygen
 docs/dev
 
-# Sublime Text settings
-*.sublime-workspace
-*.sublime-project
-
-# Eclipse Project settings
-*.*project
-
-# CMake generated files
-*.gen.cmake
+# LevelDB files
+*.sst
+*.ldb
+LOCK
+LOG*
+CURRENT
+MANIFEST-*

.travis.yml

@@ -2,15 +2,16 @@
 # one using CMake, and one using make.
 env:
   matrix:
-    - WITH_CUDA=false WITH_CMAKE=false
     - WITH_CUDA=false WITH_CMAKE=true
-    - WITH_CUDA=true WITH_CMAKE=false
     - WITH_CUDA=true WITH_CMAKE=true
 
 language: cpp
 
 # Cache Ubuntu apt packages.
-cache: apt
+cache:
+  apt: true
+  directories:
+    - /home/travis/miniconda
 
 compiler: gcc
 
@@ -22,7 +23,8 @@ install:
   - sudo -E $SCRIPTS/travis_install.sh
 
 before_script:
-  - export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib
+  - export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib:/usr/local/cuda/lib64
+  - export PATH=/home/travis/miniconda/bin:$PATH
   - if ! $WITH_CMAKE; then $SCRIPTS/travis_setup_makefile_config.sh; fi
 
 script: $SCRIPTS/travis_build_and_test.sh
@@ -34,7 +36,7 @@ notifications:
 # your Caffe fork.  To configure your git email address, use:
 #     git config --global user.email [email protected]
   email:
-    on_success: always
+    on_success: change
     on_failure: always
 
 # IRC notifications disabled by default.

CHANGELOG.md

@@ -0,0 +1,53 @@
+## Apr 27, 2016
+
+Features:
+
+  - Supported cuDNN v5
+  - Use the cuDNN's BatchNormalization implementation as the default engine for BN layer
+  - BN layer will now store running variance in its fourth blob.
+  - the script `python/bn_convert_style.py` is added to help converting the bn style forth and back.
+
+## Dec 23, 2015
+
+Features:
+
+  - Implemented a planning algorithm to globally optimize the cudnn workspace consumption and speed trade-off.
+  - Now `richness` parameter specifies the total memory in MBs available to cudnn for convolution workspaces.
+  - Now the framework will try to find the best convolution algorithm combinations under memory limit.
+
+## Dec 17, 2015
+
+Features:
+
+  - cuDNN v4 support
+  - 20% overall speed gain with faster convolution and batch normalization
+  - the native batch normalization is changed to comply with cuDNN. Use the script `python/bn_var_to_inv_std.py` to upgrade your models.
+
+## Nov 22, 2015
+
+Features:
+  - python layer can expose a prefetch() method, which will be run in parallel with network processing.
+
+## Oct 13, 2015
+
+Features:
+  - Improved cuDNN wrapper to use less GPU memory. 
+  - Now there is a new parameter `richness` which controls the limit of workspace for cuDNN.
+
+## Sep 30, 2015
+
+Features:
+  - Support for cuDNN v3.
+
+## Sep. 7, 2015
+
+Features:
+  - New mechanism for parallel comminucation reduced parallel overhead.
+  - Batch normalization, courtesy of @Cysu.
+
+## Jul, 2015
+
+Features:
+  - Action recognition tools, scripts, and examples.
+  - Basic parallel training support
+  - Various extra data augmentations

CMakeLists.txt

@@ -1,90 +1,71 @@
-cmake_minimum_required(VERSION 2.8.8)
-project( Caffe )
+cmake_minimum_required(VERSION 2.8.7)
 
-###    Build Options     ##########################################################################
+# ---[ Caffe project
+project(Caffe C CXX)
 
-option(CPU_ONLY "Build Caffe without GPU support" OFF)
-option(BUILD_PYTHON "Build Python wrapper" OFF)
-option(BUILD_MATLAB "Build Matlab wrapper" OFF)
-option(BUILD_EXAMPLES "Build examples" ON)
-option(BUILD_SHARED_LIBS "Build SHARED libs if ON and STATIC otherwise" OFF)
+# ---[ Using cmake scripts and modules
+list(APPEND CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR}/cmake/Modules)
 
-if(NOT BLAS)
-    set(BLAS atlas)
-endif()
+include(ExternalProject)
 
-if(NOT CUDA_TEST_DEVICE)
-    set(CUDA_TEST_DEVICE -1)
-endif()
+include(cmake/Utils.cmake)
+include(cmake/Targets.cmake)
+include(cmake/Misc.cmake)
+include(cmake/Summary.cmake)
+include(cmake/ConfigGen.cmake)
 
-#    Install Prefix
-if (CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT)
-    set (CMAKE_INSTALL_PREFIX "${CMAKE_BINARY_DIR}/install" CACHE PATH "Default install path" FORCE )
-endif()
+# ---[ Options
+caffe_option(CPU_ONLY  "Build Caffe without CUDA support" OFF) # TODO: rename to USE_CUDA
+caffe_option(USE_CUDNN "Build Caffe with cuDNN libary support" ON IF NOT CPU_ONLY)
+caffe_option(BUILD_SHARED_LIBS "Build shared libraries" ON)
+caffe_option(BUILD_python "Build Python wrapper" ON)
+set(python_version "2" CACHE STRING "Specify which python version to use")
+caffe_option(BUILD_matlab "Build Matlab wrapper" OFF IF UNIX OR APPLE)
+caffe_option(BUILD_docs   "Build documentation" ON IF UNIX OR APPLE)
+caffe_option(BUILD_python_layer "Build the Caffe python layer" ON)
+
+caffe_option(USE_MPI "whether to include MPI parallelization" OFF) #Used to switch on and off MPI
 
-###    Configuration    ###########################################################################
-#    Compiler Flags
-set(CMAKE_CXX_COMPILER_FLAGS ${CMAKE_CXX_COMPILER_FLAGS} -Wall)
-set(CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS} -fPIC) # set global flags
-set(CMAKE_CXX_FLAGS_DEBUG ${CMAKE_CXX_FLAGS_DEBUG}) # set debug flags
-set(CMAKE_CXX_FLAGS_RELEASE ${CMAKE_CXX_FLAGS_RELEASE}) # set release flags
+# ---[ Dependencies
+include(cmake/Dependencies.cmake)
 
-#    Global Definitions
-if(CPU_ONLY)
-    add_definitions(-DCPU_ONLY)
+# ---[ Flags
+if(UNIX OR APPLE)
+  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC -Wall")
 endif()
 
-#    Include Directories
-set(${PROJECT_NAME}_INCLUDE_DIRS ${CMAKE_SOURCE_DIR}/include)
-include_directories(${${PROJECT_NAME}_INCLUDE_DIRS})
-include_directories(${CMAKE_SOURCE_DIR}/src)
+if(USE_libstdcpp)
+  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -stdlib=libstdc++")
+  message("-- Warning: forcing libstdc++ (controlled by USE_libstdcpp option in cmake)")
+endif()
 
-#    CMake Scripts dir
-set(CMAKE_SCRIPT_DIR ${CMAKE_SOURCE_DIR}/CMakeScripts)
+add_definitions(-DGTEST_USE_OWN_TR1_TUPLE)
 
-#	CMake module path for custom module finding
-set( CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_SCRIPT_DIR})
+# ---[ Warnings
+caffe_warnings_disable(CMAKE_CXX_FLAGS -Wno-sign-compare -Wno-uninitialized)
 
-#    CUDA is required globally
-if(NOT CPU_ONLY)
-    find_package(CUDA 5.5 REQUIRED)
-    include_directories(${CUDA_INCLUDE_DIRS})
-endif()
+# ---[ Config generation
+configure_file(cmake/Templates/caffe_config.h.in "${PROJECT_BINARY_DIR}/caffe_config.h")
 
-###    Subdirectories    ##########################################################################
+# ---[ Includes
+set(Caffe_INCLUDE_DIR ${PROJECT_SOURCE_DIR}/include)
+include_directories(${Caffe_INCLUDE_DIR} ${PROJECT_BINARY_DIR})
+include_directories(BEFORE src) # This is needed for gtest.
 
+# ---[ Subdirectories
 add_subdirectory(src/gtest)
 add_subdirectory(src/caffe)
 add_subdirectory(tools)
+add_subdirectory(examples)
+add_subdirectory(python)
+add_subdirectory(matlab)
+add_subdirectory(docs)
 
-if(BUILD_EXAMPLES)
-    message(STATUS "Examples enabled")
-    add_subdirectory(examples)
-endif()
-
-if(BUILD_PYTHON)
-    message(STATUS "Python enabled")
-    add_subdirectory(python)
-endif()
-
-if(BUILD_MATLAB)
-    message(STATUS "Matlab enabled")
-    add_subdirectory(matlab)
-endif()
-
-###    Lint Target Setup    ##########################################################################
-
-set(LINT_TARGET lint)
-set(LINT_SCRIPT ${CMAKE_SCRIPT_DIR}/lint.cmake)
-add_custom_target(
-    ${LINT_TARGET} 
-    COMMAND ${CMAKE_COMMAND} -P ${LINT_SCRIPT}
-)
-
-###    Install    #################################################################################
-
-#    Install Includes
-file(GLOB folders ${${PROJECT_NAME}_INCLUDE_DIRS}/*)
-install(DIRECTORY ${folders} DESTINATION include)
+# ---[ Linter target
+add_custom_target(lint COMMAND ${CMAKE_COMMAND} -P ${PROJECT_SOURCE_DIR}/cmake/lint.cmake)
 
+# ---[ Configuration summary
+caffe_print_configuration_summary()
 
+# ---[ Export configs generation
+caffe_generate_export_configs()