mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Fixed resizing REORG_OLD layer

Browse Source
This commit is contained in:
AlexeyAB
2019-05-27 20:55:42 +03:00
parent 2b7fea761f
commit 24f7d94ab4
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/detector.c
View File

@ -86,7 +86,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
printf("\n Error: You set incorrect value batch=1 for Training! You should set batch=64 subdivision=64 \n"); printf("\n Error: You set incorrect value batch=1 for Training! You should set batch=64 subdivision=64 \n");
getchar(); getchar();
} }
else if (actual_batch_size < 64) { else if (actual_batch_size < 8) {
printf("\n Warning: You set batch=%d lower than 64! It is recommended to set batch=64 subdivision=64 \n", actual_batch_size); printf("\n Warning: You set batch=%d lower than 64! It is recommended to set batch=64 subdivision=64 \n", actual_batch_size);
} }

2
src/network.c
View File

@ -523,6 +523,8 @@ int resize_network(network *net, int w, int h)
resize_upsample_layer(&l, w, h); resize_upsample_layer(&l, w, h);
}else if(l.type == REORG){ }else if(l.type == REORG){
resize_reorg_layer(&l, w, h); resize_reorg_layer(&l, w, h);
} else if (l.type == REORG_OLD) {
resize_reorg_old_layer(&l, w, h);
}else if(l.type == AVGPOOL){ }else if(l.type == AVGPOOL){
resize_avgpool_layer(&l, w, h); resize_avgpool_layer(&l, w, h);
}else if(l.type == NORMALIZATION){ }else if(l.type == NORMALIZATION){