| | |
| | | if (ymax > h) ymax = h; |
| | | |
| | | for(j = 0; j < classes; ++j){ |
| | | int class = j; |
| | | if (probs[i][class]) fprintf(fp, "%d %d %f %f %f %f %f\n", id, j+1, probs[i][class], |
| | | int class_id = j; |
| | | if (probs[i][class_id]) fprintf(fp, "%d %d %f %f %f %f %f\n", id, j+1, probs[i][class_id], |
| | | xmin, ymin, xmax, ymax); |
| | | } |
| | | } |
| | |
| | | avg_precision += cur_precision; |
| | | } |
| | | avg_precision = avg_precision / 11; |
| | | printf("class = %d, name = %s, \t ap = %2.2f %% \n", i, names[i], avg_precision*100); |
| | | printf("class_id = %d, name = %s, \t ap = %2.2f %% \n", i, names[i], avg_precision*100); |
| | | mean_average_precision += avg_precision; |
| | | } |
| | | |
| | |
| | | |
| | | void run_detector(int argc, char **argv) |
| | | { |
| | | int http_stream_port = find_int_arg(argc, argv, "-http_port", -1); |
| | | char *out_filename = find_char_arg(argc, argv, "-out_filename", 0); |
| | | char *prefix = find_char_arg(argc, argv, "-prefix", 0); |
| | | float thresh = find_float_arg(argc, argv, "-thresh", .24); |
| | |
| | | char **names = get_labels(name_list); |
| | | if(filename) |
| | | if (filename[strlen(filename) - 1] == 0x0d) filename[strlen(filename) - 1] = 0; |
| | | demo(cfg, weights, thresh, cam_index, filename, names, classes, frame_skip, prefix, out_filename); |
| | | demo(cfg, weights, thresh, cam_index, filename, names, classes, frame_skip, prefix, out_filename, http_stream_port); |
| | | } |
| | | } |