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

support darknet-style avgpool (global avgpool)

Browse Source
This commit is contained in:
thnkinbtfly
2020-12-14 05:31:10 +00:00
parent b7af24980e
commit 2d2144639f
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/DarknetParser.cpp
+5 -1
View File
@@ -134,7 +134,11 @@ namespace tk { namespace dnn {
f.padding_x, f.padding_y, tk::dnn::POOLING_MAX)); f.padding_x, f.padding_y, tk::dnn::POOLING_MAX));
} else if(f.type == "avgpool") { } else if(f.type == "avgpool") {
netLayers.push_back(new tk::dnn::Pooling(net, f.size_x, f.size_y, f.stride_x, f.stride_y, auto output_dim = net->getOutputDim();
int stride = 1;
assert(f.padding_x == 0 && f.padding_y == 0);
netLayers.push_back(new tk::dnn::Pooling(net, output_dim.h, output_dim.w, stride, stride,
f.padding_x, f.padding_y, tk::dnn::POOLING_AVERAGE)); f.padding_x, f.padding_y, tk::dnn::POOLING_AVERAGE));
} else if(f.type == "shortcut") { } else if(f.type == "shortcut") {