|
|
@ -32,6 +32,7 @@ |
|
|
|
#include "dropout_layer.h" |
|
|
|
#include "dropout_layer.h" |
|
|
|
#include "route_layer.h" |
|
|
|
#include "route_layer.h" |
|
|
|
#include "shortcut_layer.h" |
|
|
|
#include "shortcut_layer.h" |
|
|
|
|
|
|
|
#include "scale_channels_layer.h" |
|
|
|
#include "yolo_layer.h" |
|
|
|
#include "yolo_layer.h" |
|
|
|
#include "upsample_layer.h" |
|
|
|
#include "upsample_layer.h" |
|
|
|
#include "parser.h" |
|
|
|
#include "parser.h" |
|
|
@ -523,6 +524,8 @@ int resize_network(network *net, int w, int h) |
|
|
|
resize_route_layer(&l, net); |
|
|
|
resize_route_layer(&l, net); |
|
|
|
}else if (l.type == SHORTCUT) { |
|
|
|
}else if (l.type == SHORTCUT) { |
|
|
|
resize_shortcut_layer(&l, w, h); |
|
|
|
resize_shortcut_layer(&l, w, h); |
|
|
|
|
|
|
|
}else if (l.type == SCALE_CHANNELS) { |
|
|
|
|
|
|
|
resize_scale_channels_layer(&l, w, h); |
|
|
|
}else if (l.type == UPSAMPLE) { |
|
|
|
}else if (l.type == UPSAMPLE) { |
|
|
|
resize_upsample_layer(&l, w, h); |
|
|
|
resize_upsample_layer(&l, w, h); |
|
|
|
}else if(l.type == REORG){ |
|
|
|
}else if(l.type == REORG){ |
|
|
|