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

Extracting features from VOC with temp filters

Browse Source
This commit is contained in:
Joseph Redmon
2014-02-14 16:09:07 -08:00
parent 118bdd6f62
commit 228d3663f8
9 changed files with 128 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/network.c
View File

@ -331,6 +331,34 @@ int get_network_output_size_layer(network net, int i)
return 0;
}
int reset_network_size(network net, int h, int w, int c)
{
int i;
for (i = 0; i < net.n; ++i){
if(net.types[i] == CONVOLUTIONAL){
convolutional_layer *layer = (convolutional_layer *)net.layers[i];
layer->h = h;
layer->w = w;
layer->c = c;
image output = get_convolutional_image(*layer);
h = output.h;
w = output.w;
c = output.c;
}
else if(net.types[i] == MAXPOOL){
maxpool_layer *layer = (maxpool_layer *)net.layers[i];
layer->h = h;
layer->w = w;
layer->c = c;
image output = get_maxpool_image(*layer);
h = output.h;
w = output.w;
c = output.c;
}
}
return 0;
}
int get_network_output_size(network net)
{
int i = net.n-1;