mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

ZED 3D Camera support added to ./uselib (yolo_console_cpp.exe) example

Browse Source
This commit is contained in:
AlexeyAB
2019-03-18 02:48:52 +03:00
parent 7a854302ef
commit b6e15f1656
65 changed files with 1324 additions and 662 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/convolutional_kernels.cu
View File

@ -2,12 +2,6 @@
#include "curand.h"
#include "cublas_v2.h"
#ifdef CUDNN
#ifndef USE_CMAKE_LIBS
#pragma comment(lib, "cudnn.lib")
#endif
#endif
#include "convolutional_layer.h"
#include "batchnorm_layer.h"
#include "gemm.h"
@ -15,7 +9,7 @@
#include "im2col.h"
#include "col2im.h"
#include "utils.h"
#include "cuda.h"
#include "dark_cuda.h"
__global__ void binarize_kernel(float *x, int n, float *binary)
@ -598,6 +592,7 @@ void forward_convolutional_layer_gpu(convolutional_layer l, network_state state)
void backward_convolutional_layer_gpu(convolutional_layer l, network_state state)
{
if(state.net.try_fix_nan) constrain_ongpu(l.outputs*l.batch, 1, l.delta_gpu, 1);
gradient_array_ongpu(l.output_gpu, l.outputs*l.batch, l.activation, l.delta_gpu);
if (!l.batch_normalize)