OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Added network resizing for SAM layer

This commit is contained in:
AlexeyAB 2020-01-09 19:44:08 +03:00
parent a66bc8d6ff
commit 283333b52f
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/network.c
View File

@ -541,6 +541,8 @@ int resize_network(network *net, int w, int h)
resize_shortcut_layer(&l, w, h, net);
}else if (l.type == SCALE_CHANNELS) {
resize_scale_channels_layer(&l, net);
}else if (l.type == SAM) {
resize_sam_layer(&l, w, h);
}else if (l.type == DROPOUT) {
resize_dropout_layer(&l, inputs);
l.out_w = l.w = w;