Open Yeong-Seo opened 4 years ago
Hi
I have added source in detector.c for loop.
I use detect image repeatably.
This is added code.
void test_4picture(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh, float hier_thresh, int dont_show, int ext_output, int save_labels, char *outfile, int letter_box) { FILE *stream; char memBuff[255]; char filepath[30] = {'\0', }; int k = 0; char *strtokk; list *options = read_data_cfg(datacfg); char *name_list = option_find_str(options, "names", "data/names.list"); int names_size = 0; char **names = get_labels_custom(name_list, &names_size); //get_labels(name_list); image **alphabet = load_alphabet(); network net = parse_network_cfg_custom(cfgfile, 1, 1); // set batch=1 if (weightfile) { load_weights(&net, weightfile); } fuse_conv_batchnorm(net); calculate_binary_weights(net); if (net.layers[net.n - 1].classes != names_size) { printf(" Error: in the file %s number of names %d that isn't equal to classes=%d in the file %s \n", name_list, names_size, net.layers[net.n - 1].classes, cfgfile); if (net.layers[net.n - 1].classes > names_size) getchar(); } srand(2222222); char buff[256]; char *input = buff; char *json_buf = NULL; int json_image_id = 0; FILE* json_file = NULL; if (outfile) { json_file = fopen(outfile, "wb"); char *tmp = "[\n"; fwrite(tmp, sizeof(char), strlen(tmp), json_file); } int j; float nms = .45; // 0.4F while (1) { k++; if(k == 5) { k = 1; syslog(LOG_INFO | LOG_LOCAL1, "Deep Learning Stop"); system("sudo pkill -STOP -ef 'sudo sh picture_run.sh'"); } sprintf(input, "%s%d%s", "/media/ramdisk/", k, ".jpg"); image im = load_image(input, 0, 0, net.c); image sized; if(letter_box) sized = letterbox_image(im, net.w, net.h); else sized = resize_image(im, net.w, net.h); layer l = net.layers[net.n - 1]; //box *boxes = calloc(l.w*l.h*l.n, sizeof(box)); //float **probs = calloc(l.w*l.h*l.n, sizeof(float*)); //for(j = 0; j < l.w*l.h*l.n; ++j) probs[j] = (float*)calloc(l.classes, sizeof(float)); float *X = sized.data; //time= what_time_is_it_now(); double time = get_time_point(); network_predict(net, X); //network_predict_image(&net, im); letterbox = 1; printf("%s: Predicted in %lf seconds.\n", input, ((double)get_time_point() - time) / 1000000); //printf("%s: Predicted in %f seconds.\n", input, (what_time_is_it_now()-time)); int nboxes = 0; detection *dets = get_network_boxes(&net, im.w, im.h, thresh, hier_thresh, 0, 1, &nboxes, letter_box); if (nms) do_nms_sort(dets, nboxes, l.classes, nms); draw_detections_v3(im, dets, nboxes, thresh, names, alphabet, l.classes, ext_output); //save_image(im, "predictions"); if (!dont_show) { show_image(im, "predictions"); } if (outfile) { if (json_buf) { char *tmp = ", \n"; fwrite(tmp, sizeof(char), strlen(tmp), json_file); } ++json_image_id; json_buf = detection_to_json(dets, nboxes, l.classes, names, json_image_id, input); fwrite(json_buf, sizeof(char), strlen(json_buf), json_file); free(json_buf); } // pseudo labeling concept - fast.ai if (save_labels) { char labelpath[4096]; replace_image_to_label(input, labelpath); FILE* fw = fopen(labelpath, "wb"); int i; for (i = 0; i < nboxes; ++i) { char buff[1024]; int class_id = -1; float prob = 0; for (j = 0; j < l.classes; ++j) { if (dets[i].prob[j] > thresh && dets[i].prob[j] > prob) { prob = dets[i].prob[j]; class_id = j; } } if (class_id >= 0) { sprintf(buff, "%d %2.4f %2.4f %2.4f %2.4f\n", class_id, dets[i].bbox.x, dets[i].bbox.y, dets[i].bbox.w, dets[i].bbox.h); fwrite(buff, sizeof(char), strlen(buff), fw); } } fclose(fw); } free_detections(dets, nboxes); free_image(im); free_image(sized); if (!dont_show) { wait_until_press_key_cv(); destroy_all_windows_cv(); } //destroy_all_windows_cv(); if (filename && !strstr(filename, "camera")) break; } if (outfile) { char *tmp = "\n]"; fwrite(tmp, sizeof(char), strlen(tmp), json_file); fclose(json_file); } // free memory free_ptrs((void**)names, net.layers[net.n - 1].classes); free_list_contents_kvp(options); 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); }
It's the same as void test_detector(). just add loop and pkill -STOP code.
It works well at first, but not later (more and more repeat). error message is Cannot load image.
Is there a problem with my code?
at first
later
similar problem
Hi
I have added source in detector.c for loop.
I use detect image repeatably.
This is added code.
It's the same as void test_detector(). just add loop and pkill -STOP code.
It works well at first, but not later (more and more repeat). error message is Cannot load image.
Is there a problem with my code?
at first
later