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

add scale_channels, swish, logistic(sigmoid), avgpool ; add enet_coco(EfficientNetB0-Yolov3).cfg #172

Open
thnkinbtfly wants to merge 6 commits from thnkinbtfly/master into master
pull from: thnkinbtfly/master
merge into: mmr:master
Conversation 0 Commits 6 Files Changed 13
+5 -1
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 2d2144639f - Show all commits
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));
} 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));
} else if(f.type == "shortcut") {