Attempt at visualizing ImageNet Features

This commit is contained in:
Joseph Redmon
2014-04-11 01:00:27 -07:00
parent 2ea63c0e99
commit cc06817efa
14 changed files with 737 additions and 92 deletions

View File

@ -428,13 +428,14 @@ image get_network_image(network net)
void visualize_network(network net)
{
image *prev = 0;
int i;
char buff[256];
for(i = 0; i < net.n; ++i){
sprintf(buff, "Layer %d", i);
if(net.types[i] == CONVOLUTIONAL){
convolutional_layer layer = *(convolutional_layer *)net.layers[i];
visualize_convolutional_layer(layer, buff);
prev = visualize_convolutional_layer(layer, buff, prev);
}
}
}
@ -506,3 +507,4 @@ float network_accuracy(network net, data d)
return acc;
}