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

detection better?

Browse Source
This commit is contained in:
Joseph Redmon
2015-04-07 15:25:30 -07:00
parent a05c4bd2e9
commit 1fd10265f8
5 changed files with 65 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/network.c
View File

@ -500,6 +500,18 @@ int get_network_input_size(network net)
return get_network_input_size_layer(net, 0);
}
detection_layer *get_network_detection_layer(network net)
{
int i;
for(i = 0; i < net.n; ++i){
if(net.types[i] == DETECTION){
detection_layer *layer = (detection_layer *)net.layers[i];
return layer;
}
}
return 0;
}
image get_network_image_layer(network net, int i)
{
if(net.types[i] == CONVOLUTIONAL){