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){