adding reorg3d layer

2020-06-14 18:48:25 +04:30
parent b8ecd3c0cb
commit e20d3a0861
1 changed files with 1 additions and 1 deletions
@@ -182,7 +182,7 @@ namespace tk { namespace dnn {
            }
            netLayers.push_back(new tk::dnn::Route(net, layers.data(), layers.size()));

-        } else if(f.type == "reorg") {
+        } else if(f.type == "reorg" || f.type == "reorg3d") {
            netLayers.push_back(new tk::dnn::Reorg(net, f.stride_x));

        } else if(f.type == "region") {