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

1
src/yolo_v2_class.cpp
View File

@ -69,6 +69,7 @@ YOLODLL_API Detector::Detector(std::string cfg_filename, std::string weight_file
}
set_batch_network(&net, 1);
net.gpu_index = cur_gpu_id;
fuse_conv_batchnorm(net);
layer l = net.layers[net.n - 1];
int j;