| | |
| | | for(c = 0; c < side; ++c){ |
| | | j = (r*side + c) * elems; |
| | | int class = max_index(box+j, classes); |
| | | if(box[j+class] > .2){ |
| | | if(box[j+class] > 0){ |
| | | printf("%f %s\n", box[j+class], class_names[class]); |
| | | float red = get_color(0,class,classes); |
| | | float green = get_color(1,class,classes); |
| | |
| | | //plist = get_paths("/home/pjreddie/data/voc/no_2012_val.txt"); |
| | | //plist = get_paths("/home/pjreddie/data/voc/no_2007_test.txt"); |
| | | //plist = get_paths("/home/pjreddie/data/voc/val_2012.txt"); |
| | | plist = get_paths("/home/pjreddie/data/voc/no_2007_test.txt"); |
| | | //plist = get_paths("/home/pjreddie/data/voc/no_2007_test.txt"); |
| | | //plist = get_paths("/home/pjreddie/data/coco/trainval.txt"); |
| | | //plist = get_paths("/home/pjreddie/data/voc/all2007-2012.txt"); |
| | | plist = get_paths("/home/pjreddie/data/voc/all2007-2012.txt"); |
| | | } |
| | | paths = (char **)list_to_array(plist); |
| | | pthread_t load_thread = load_data_detection_thread(imgs, paths, plist->size, classes, net.w, net.h, side, side, background, &buffer); |
| | |
| | | clock_t time; |
| | | char filename[256]; |
| | | while(1){ |
| | | printf("Image Path: "); |
| | | fflush(stdout); |
| | | fgets(filename, 256, stdin); |
| | | strtok(filename, "\n"); |
| | | image im = load_image_color(filename,0,0); |
| | | image sized = resize_image(im, im_size, im_size); |
| | | printf("%d %d %d\n", im.h, im.w, im.c); |
| | | float *X = sized.data; |
| | | time=clock(); |
| | | float *predictions = network_predict(net, X); |