Fixed scale_channels_layer resize

6 years ago · 5a6afe96d3
parent 54e2d0b0e8
commit 5a6afe96d3
1 changed files with 3 additions and 0 deletions
--- 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){