From 5a6afe96d3aa8aed19405577db7dba0ff173c848 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Wed, 3 Jul 2019 13:54:13 +0300 Subject: [PATCH] Fixed scale_channels_layer resize --- src/network.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/network.c b/src/network.c index e0be3995..faf3db98 100644 --- a/src/network.c +++ b/src/network.c @@ -32,6 +32,7 @@ #include "dropout_layer.h" #include "route_layer.h" #include "shortcut_layer.h" +#include "scale_channels_layer.h" #include "yolo_layer.h" #include "upsample_layer.h" #include "parser.h" @@ -523,6 +524,8 @@ int resize_network(network *net, int w, int h) resize_route_layer(&l, net); }else if (l.type == SHORTCUT) { resize_shortcut_layer(&l, w, h); + }else if (l.type == SCALE_CHANNELS) { + resize_scale_channels_layer(&l, w, h); }else if (l.type == UPSAMPLE) { resize_upsample_layer(&l, w, h); }else if(l.type == REORG){