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

Added Focal Loss to yolo-layer

Browse Source
This commit is contained in:
AlexeyAB
2018-04-05 23:27:02 +03:00
parent be9d971ddb
commit 943f6e874b
5 changed files with 51 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/network.c
View File

@ -757,7 +757,7 @@ void fuse_conv_batchnorm(network net)
layer *l = &net.layers[j];
if (l->type == CONVOLUTIONAL) {
printf(" Fuse Convolutional layer \t\t l->size = %d \n", l->size);
//printf(" Merges Convolutional-%d and batch_norm \n", j);
if (l->batch_normalize) {
int f;
@ -783,7 +783,7 @@ void fuse_conv_batchnorm(network net)
}
}
else {
printf(" Skip layer: %d \n", l->type);
//printf(" Fusion skip layer type: %d \n", l->type);
}
}
}