From c62b4f35aa2c59d7db0fd177affeed14b1ba4bcb Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 08 Sep 2016 07:04:39 +0000
Subject: [PATCH] adding coco models
---
src/convolutional_kernels.cu | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/convolutional_kernels.cu b/src/convolutional_kernels.cu
index 2376835..43b3f9a 100644
--- a/src/convolutional_kernels.cu
+++ b/src/convolutional_kernels.cu
@@ -17,7 +17,7 @@
{
int i = (blockIdx.x + blockIdx.y*gridDim.x) * blockDim.x + threadIdx.x;
if (i >= n) return;
- binary[i] = (x[i] > 0) ? 1 : -1;
+ binary[i] = (x[i] >= 0) ? 1 : -1;
}
void binarize_gpu(float *x, int n, float *binary)
@@ -60,6 +60,7 @@
mean = mean / size;
for(i = 0; i < size; ++i){
binary[f*size + i] = (filters[f*size + i] > 0) ? mean : -mean;
+ //binary[f*size + i] = filters[f*size + i];
}
}
@@ -71,8 +72,6 @@
void forward_convolutional_layer_gpu(convolutional_layer l, network_state state)
{
- int i;
-
fill_ongpu(l.outputs*l.batch, 0, l.output_gpu, 1);
if(l.binary){
binarize_filters_gpu(l.filters_gpu, l.n, l.c*l.size*l.size, l.binary_filters_gpu);
@@ -103,6 +102,7 @@
l.output_gpu);
#else
+ int i;
int m = l.n;
int k = l.size*l.size*l.c;
int n = l.out_w*l.out_h;
--
Gitblit v1.10.0