Fixed scale_channels_layer resize

This commit is contained in:
AlexeyAB
2019-07-03 13:54:13 +03:00
parent 54e2d0b0e8
commit 5a6afe96d3

View File

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