| | |
|---|
| | | #ifdef _DEBUG |
|---|
| | | #include <stdlib.h> |
|---|
| | | #include <crtdbg.h> |
|---|
| | | #endif |
|---|
| | | |
|---|
| | | #include "network.h" |
|---|
| | | #include "region_layer.h" |
|---|
| | | #include "cost_layer.h" |
|---|
| | |
|---|
| | | args.n = imgs; |
|---|
| | | args.m = plist->size; |
|---|
| | | args.classes = classes; |
|---|
| | | args.flip = net.flip; |
|---|
| | | args.jitter = jitter; |
|---|
| | | args.num_boxes = l.max_boxes; |
|---|
| | | args.small_object = net.small_object; |
|---|
| | |
|---|
| | | truth_dif = read_boxes(labelpath_dif, &num_labels_dif); |
|---|
| | | } |
|---|
| | | |
|---|
| | | const int checkpoint_detections_count = detections_count; |
|---|
| | | |
|---|
| | | for (i = 0; i < nboxes; ++i) { |
|---|
| | | |
|---|
| | | int class_id; |
|---|
| | |
|---|
| | | |
|---|
| | | // calc avg IoU, true-positives, false-positives for required Threshold |
|---|
| | | if (prob > thresh_calc_avg_iou) { |
|---|
| | | if (truth_index > -1) { |
|---|
| | | int z, found = 0; |
|---|
| | | for (z = checkpoint_detections_count; z < detections_count-1; ++z) |
|---|
| | | if (detections[z].unique_truth_index == truth_index) { |
|---|
| | | found = 1; break; |
|---|
| | | } |
|---|
| | | |
|---|
| | | if(truth_index > -1 && found == 0) { |
|---|
| | | avg_iou += max_iou; |
|---|
| | | ++tp_for_thresh; |
|---|
| | | } |
|---|
| | |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | avg_iou = avg_iou / (tp_for_thresh + fp_for_thresh); |
|---|
| | | if((tp_for_thresh + fp_for_thresh) > 0) |
|---|
| | | avg_iou = avg_iou / (tp_for_thresh + fp_for_thresh); |
|---|
| | | |
|---|
| | | |
|---|
| | | // SORT(detections) |
|---|
| | |
|---|
| | | while(1){ |
|---|
| | | if(filename){ |
|---|
| | | strncpy(input, filename, 256); |
|---|
| | | if (input[strlen(input) - 1] == 0x0d) input[strlen(input) - 1] = 0; |
|---|
| | | if(strlen(input) > 0) |
|---|
| | | if (input[strlen(input) - 1] == 0x0d) input[strlen(input) - 1] = 0; |
|---|
| | | } else { |
|---|
| | | printf("Enter Image Path: "); |
|---|
| | | fflush(stdout); |
|---|
| | |
|---|
| | | } |
|---|
| | | image im = load_image_color(input,0,0); |
|---|
| | | int letterbox = 0; |
|---|
| | | image sized = resize_image(im, net.w, net.h); |
|---|
| | | //image sized = letterbox_image(im, net.w, net.h); letterbox = 1; |
|---|
| | | //image sized = resize_image(im, net.w, net.h); |
|---|
| | | image sized = letterbox_image(im, net.w, net.h); letterbox = 1; |
|---|
| | | layer l = net.layers[net.n-1]; |
|---|
| | | |
|---|
| | | //box *boxes = calloc(l.w*l.h*l.n, sizeof(box)); |
|---|
| | |
|---|
| | | |
|---|
| | | float *X = sized.data; |
|---|
| | | time= what_time_is_it_now(); |
|---|
| | | //network_predict(net, X); |
|---|
| | | network_predict_image(&net, im); letterbox = 1; |
|---|
| | | network_predict(net, X); |
|---|
| | | //network_predict_image(&net, im); letterbox = 1; |
|---|
| | | printf("%s: Predicted in %f seconds.\n", input, (what_time_is_it_now()-time)); |
|---|
| | | //get_region_boxes(l, 1, 1, thresh, probs, boxes, 0, 0); |
|---|
| | | // if (nms) do_nms_sort_v2(boxes, probs, l.w*l.h*l.n, l.classes, nms); |
|---|
| | |
|---|
| | | #endif |
|---|
| | | if (filename) break; |
|---|
| | | } |
|---|
| | | |
|---|
| | | // free memory |
|---|
| | | free_ptrs(names, net.layers[net.n - 1].classes); |
|---|
| | | free_list(options); |
|---|
| | | |
|---|
| | | int i; |
|---|
| | | const int nsize = 8; |
|---|
| | | for (j = 0; j < nsize; ++j) { |
|---|
| | | for (i = 32; i < 127; ++i) { |
|---|
| | | free_image(alphabet[j][i]); |
|---|
| | | } |
|---|
| | | free(alphabet[j]); |
|---|
| | | } |
|---|
| | | free(alphabet); |
|---|
| | | |
|---|
| | | free_network(net); |
|---|
| | | } |
|---|
| | | |
|---|
| | | void run_detector(int argc, char **argv) |
|---|
| | |
|---|
| | | char *cfg = argv[4]; |
|---|
| | | char *weights = (argc > 5) ? argv[5] : 0; |
|---|
| | | if(weights) |
|---|
| | | if (weights[strlen(weights) - 1] == 0x0d) weights[strlen(weights) - 1] = 0; |
|---|
| | | if(strlen(weights) > 0) |
|---|
| | | if (weights[strlen(weights) - 1] == 0x0d) weights[strlen(weights) - 1] = 0; |
|---|
| | | char *filename = (argc > 6) ? argv[6]: 0; |
|---|
| | | if(0==strcmp(argv[2], "test")) test_detector(datacfg, cfg, weights, filename, thresh, hier_thresh, dont_show); |
|---|
| | | else if(0==strcmp(argv[2], "train")) train_detector(datacfg, cfg, weights, gpus, ngpus, clear, dont_show); |
|---|
| | |
|---|
| | | char *name_list = option_find_str(options, "names", "data/names.list"); |
|---|
| | | char **names = get_labels(name_list); |
|---|
| | | if(filename) |
|---|
| | | if (filename[strlen(filename) - 1] == 0x0d) filename[strlen(filename) - 1] = 0; |
|---|
| | | if(strlen(filename) > 0) |
|---|
| | | if (filename[strlen(filename) - 1] == 0x0d) filename[strlen(filename) - 1] = 0; |
|---|
| | | demo(cfg, weights, thresh, hier_thresh, cam_index, filename, names, classes, frame_skip, prefix, out_filename, |
|---|
| | | http_stream_port, dont_show); |
|---|
| | | } |
|---|
