cudnn support

unsky · Aug 22, 2018 · dccc8f6 · dccc8f6
1 parent 7d2a1ab
commit dccc8f6
Show file tree

Hide file tree

Showing 48 changed files with 5,377 additions and 264 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -29,6 +29,10 @@ include_directories(${GLOG_INCLUDE_DIR})
 link_directories(${GLOG_LIBRARY_DIR})
 file(GLOB GLOG_LIBRARIES ${GLOG_LIBRARY_DIR}/*.a)
 
+include_directories(/home/cs/cudnn/cudnn_v7/cuda/include)
+link_directories(/home/cs/cudnn/cudnn_v7/cuda/lib64)
+
+
 include_directories(/home/cs/caffe_third/for_x86/protobuf/include)
 link_directories(/home/cs/caffe_third/for_x86/protobuf/lib)
 execute_process(COMMAND protoc --cpp_out=${PROJECT_SOURCE_DIR}/src/sita/proto/ --proto_path=${PROJECT_SOURCE_DIR}/src/sita/proto/

diff --git a/README.md b/README.md
@@ -1,5 +1,5 @@
 <div align=center>
-<img src="sita.png"/>
+<img src="docs/sita.png"/>
 </div>
 
 Sita is a lightweight tracking architecture.
@@ -16,11 +16,11 @@ Integrate an trackor using sita in your project!
 # deep learning module
 
 <div align=center>
-<img src="framework.png"/>
+<img src="docs/framework.png"/>
 </div>
 
 ## Graph builder
 
 <div align=center>
-<img src="graph.gif"/>
-</div>
+<img src="docs/graph.gif"/>
+</div>