mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

tiny yolo not working

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-07 15:05:48 +02:00
parent 34198a4e8d
commit 7d570c0df4
6 changed files with 112 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Pooling.cpp
+9 -10
View File
@@ -5,20 +5,18 @@
namespace tkDNN {
Pooling::Pooling( Network *net, int winH, int winW,
int strideH, int strideW, tkdnnPoolingMode_t pool_mode) :
Pooling::Pooling( Network *net, int winH, int winW, int strideH, int strideW,
tkdnnPoolingMode_t pool_mode) :
Layer(net) {
if(winH != strideH || winW != strideW)
FatalError("stride pooling not yet implemented");
this->winH = winH;
this->winW = winW;
this->strideH = strideH;
this->strideW = strideW;
this->pool_mode = pool_mode;
this->paddingH = 0;
this->paddingW = 0;
checkCUDNN( cudnnCreatePoolingDescriptor(&poolingDesc) );
int n = input_dim.n;
@@ -46,12 +44,13 @@ Pooling::Pooling( Network *net, int winH, int winW,
net->tensorFormat, net->dataType, n, c, h, w) );
//get out dim
h = h / winH; w = w / winW;
checkCUDNN( cudnnGetPooling2dForwardOutputDim(poolingDesc, srcTensorDesc, &n, &c, &h, &w));
//h = (h + winH*this->paddingH)/strideH;
//w = (w + winW*this->paddingW)/strideW;
checkCUDNN( cudnnSetTensor4dDescriptor(dstTensorDesc,
net->tensorFormat, net->dataType, n, c, h, w) );
output_dim.n = n;
output_dim.c = c;
output_dim.h = h;