yolo layers

2017-08-01 16:08:56 +02:00
parent 8e4b3c6c17
commit b94931f9f7
21 changed files with 522 additions and 84 deletions
@@ -5,12 +5,16 @@ project (tkDNN)
 find_package(CUDA QUIET REQUIRED)

 cuda_include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include ${CUDA_INCLUDE_DIRS})
-cuda_add_library(kernels SHARED src/kernels/activation_elu.cu)
+cuda_add_library(kernels SHARED src/kernels/activation_elu.cu 
+                                src/kernels/activation_leaky.cu
+                                src/kernels/activation_logistic.cu
+                                src/kernels/reorg.cu
+                                src/kernels/softmax.cu)

 include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include ${CUDA_INCLUDE_DIRS})
 add_library(tkDNN SHARED src/Layer.cpp src/LayerWgs.cpp
    src/Dense.cpp src/Activation.cpp src/Conv2d.cpp src/Flatten.cpp src/MulAdd.cpp src/Pooling.cpp src/Softmax.cpp
-    src/Network.cpp src/utils.cpp)
+    src/Route.cpp src/Reorg.cpp src/Region.cpp src/Network.cpp src/utils.cpp)
 target_link_libraries(tkDNN kernels ${CUDA_LIBRARIES} ${CUDA_CUBLAS_LIBRARIES} -lcudnn)

 add_executable(test_simple tests/test/test.cpp)