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

LRNorm layer, Viz, better cfg

Browse Source
This commit is contained in:
Joseph Redmon 2014-04-17 09:51:38 -07:00
parent 738cd4c2d7
commit 5e468d1c13
1 changed files with 37 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
src/tests.c
View File

@ -548,7 +548,9 @@ void visualize_imagenet_topk(char *filename)
score[i] = calloc(topk, sizeof(float)); score[i] = calloc(topk, sizeof(float));
} }
int count = 0;
while(n){ while(n){
++count;
char *image_path = (char *)n->val; char *image_path = (char *)n->val;
image im = load_image(image_path, 0, 0); image im = load_image(image_path, 0, 0);
n = n->next; n = n->next;
@ -560,37 +562,46 @@ void visualize_imagenet_topk(char *filename)
forward_network(net, im.data); forward_network(net, im.data);
image out = get_network_image(net); image out = get_network_image(net);
int dh = (im.h - h)/h; int dh = (im.h - h)/(out.h-1);
int dw = (im.w - w)/w; int dw = (im.w - w)/(out.w-1);
for(i = 0; i < out.h; ++i){ //printf("%d %d\n", dh, dw);
for(j = 0; j < out.w; ++j){ for(k = 0; k < out.c; ++k){
image sub = get_sub_image(im, dh*i, dw*j, h, w); float topv = 0;
for(k = 0; k < out.c; ++k){ int topi = -1;
int topj = -1;
for(i = 0; i < out.h; ++i){
for(j = 0; j < out.w; ++j){
float val = get_pixel(out, i, j, k); float val = get_pixel(out, i, j, k);
//printf("%f, ", val); if(val > topv){
image sub_c = copy_image(sub); topv = val;
for(l = 0; l < topk; ++l){ topi = i;
if(val > score[k][l]){ topj = j;
float swap = score[k][l]; }
score[k][l] = val; }
val = swap; }
if(topv){
image swapi = vizs[k][l]; image sub = get_sub_image(im, dh*topi, dw*topj, h, w);
vizs[k][l] = sub_c; for(l = 0; l < topk; ++l){
sub_c = swapi; if(topv > score[k][l]){
} float swap = score[k][l];
score[k][l] = topv;
topv = swap;
image swapi = vizs[k][l];
vizs[k][l] = sub;
sub = swapi;
} }
free_image(sub_c);
} }
free_image(sub); free_image(sub);
} }
} }
free_image(im); free_image(im);
//printf("\n"); if(count%50 == 0){
image grid = grid_images(vizs, num, topk); image grid = grid_images(vizs, num, topk);
show_image(grid, "IMAGENET Visualization"); //show_image(grid, "IMAGENET Visualization");
save_image(grid, "IMAGENET Grid"); save_image(grid, "IMAGENET Grid Single Nonorm");
free_image(grid); free_image(grid);
}
} }
//cvWaitKey(0); //cvWaitKey(0);
} }
@ -644,7 +655,7 @@ void visualize_cat()
printf("Processing %dx%d image\n", im.h, im.w); printf("Processing %dx%d image\n", im.h, im.w);
resize_network(net, im.h, im.w, im.c); resize_network(net, im.h, im.w, im.c);
forward_network(net, im.data); forward_network(net, im.data);
image out = get_network_image(net); image out = get_network_image(net);
visualize_network(net); visualize_network(net);
cvWaitKey(1000); cvWaitKey(1000);
@ -778,7 +789,7 @@ int main(int argc, char *argv[])
//features_VOC_image(argv[1], argv[2], argv[3]); //features_VOC_image(argv[1], argv[2], argv[3]);
//features_VOC_image_size(argv[1], atoi(argv[2]), atoi(argv[3])); //features_VOC_image_size(argv[1], atoi(argv[2]), atoi(argv[3]));
//visualize_imagenet_features("data/assira/train.list"); //visualize_imagenet_features("data/assira/train.list");
visualize_imagenet_topk("data/VOC2011.list"); visualize_imagenet_topk("data/VOC2012.list");
//visualize_cat(); //visualize_cat();
//flip_network(); //flip_network();
//test_visualize(); //test_visualize();