From e20d3a086189fb48e3ba1be4eb75537b8dac31fd Mon Sep 17 00:00:00 2001 From: MohammadReza Hosseini Date: Sun, 14 Jun 2020 18:48:25 +0430 Subject: [PATCH] adding reorg3d layer --- include/tkDNN/DarknetParser.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/tkDNN/DarknetParser.h b/include/tkDNN/DarknetParser.h index cbeed48..b4c9253 100644 --- a/include/tkDNN/DarknetParser.h +++ b/include/tkDNN/DarknetParser.h @@ -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") {