mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Fixed scale_channels_layer resize
This commit is contained in:
@ -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){
|
||||
|
Reference in New Issue
Block a user