From c1bb8c129d7d4d8fc85348d5e7ac5441fa36ead5 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sat, 19 May 2018 13:52:05 +0000
Subject: [PATCH] Fixed xnor for random=1
---
src/detector.c | 39 +++++++++++++++++++++++++++++++++++++--
1 files changed, 37 insertions(+), 2 deletions(-)
diff --git a/src/detector.c b/src/detector.c
index de4bc90..3b486fb 100644
--- a/src/detector.c
+++ b/src/detector.c
@@ -132,6 +132,8 @@
int random_val = rand() % 12;
int dim_w = (random_val + (init_w / 32 - 5)) * 32; // +-160
int dim_h = (random_val + (init_h / 32 - 5)) * 32; // +-160
+ if (dim_w < 32) dim_w = 32;
+ if (dim_h < 32) dim_h = 32;
printf("%d x %d \n", dim_w, dim_h);
args.w = dim_w;
@@ -425,7 +427,7 @@
fprintf(fp, "\n]\n");
fclose(fp);
}
- fprintf(stderr, "Total Detection Time: %f Seconds\n", time(0) - start);
+ fprintf(stderr, "Total Detection Time: %f Seconds\n", (double)time(0) - start);
}
void validate_detector_recall(char *datacfg, char *cfgfile, char *weightfile)
@@ -479,6 +481,7 @@
find_replace(labelpath, ".bmp", ".txt", labelpath);
find_replace(labelpath, ".JPG", ".txt", labelpath);
find_replace(labelpath, ".JPEG", ".txt", labelpath);
+ find_replace(labelpath, ".ppm", ".txt", labelpath);
int num_labels = 0;
box_label *truth = read_boxes(labelpath, &num_labels);
@@ -636,6 +639,7 @@
find_replace(labelpath, ".bmp", ".txt", labelpath);
find_replace(labelpath, ".JPG", ".txt", labelpath);
find_replace(labelpath, ".JPEG", ".txt", labelpath);
+ find_replace(labelpath, ".ppm", ".txt", labelpath);
int num_labels = 0;
box_label *truth = read_boxes(labelpath, &num_labels);
int i, j;
@@ -899,6 +903,7 @@
find_replace(labelpath, ".bmp", ".txt", labelpath);
find_replace(labelpath, ".JPG", ".txt", labelpath);
find_replace(labelpath, ".JPEG", ".txt", labelpath);
+ find_replace(labelpath, ".ppm", ".txt", labelpath);
int num_labels = 0;
box_label *truth = read_boxes(labelpath, &num_labels);
//printf(" new path: %s \n", labelpath);
@@ -1104,12 +1109,42 @@
detection *dets = get_network_boxes(&net, im.w, im.h, thresh, hier_thresh, 0, 1, &nboxes, letterbox);
if (nms) do_nms_sort(dets, nboxes, l.classes, nms);
draw_detections_v3(im, dets, nboxes, thresh, names, alphabet, l.classes, ext_output);
- free_detections(dets, nboxes);
save_image(im, "predictions");
if (!dont_show) {
show_image(im, "predictions");
}
+/*
+ // pseudo labeling concept - fast.ai
+ {
+ char labelpath[4096];
+ find_replace(input, ".jpg", ".txt", labelpath);
+ find_replace(labelpath, ".png", ".txt", labelpath);
+ find_replace(labelpath, ".bmp", ".txt", labelpath);
+ find_replace(labelpath, ".JPG", ".txt", labelpath);
+ find_replace(labelpath, ".JPEG", ".txt", labelpath);
+ find_replace(labelpath, ".ppm", ".txt", 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);
//free(boxes);
--
Gitblit v1.10.0