From d7a30ada7e82daf7ce5ca91acac15c1723188fc5 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 16 Jan 2017 09:51:42 +0000
Subject: [PATCH] Fixed behavior if missing library cudnn.lib
---
src/data.c | 16 ++++++++++++++--
1 files changed, 14 insertions(+), 2 deletions(-)
diff --git a/src/data.c b/src/data.c
index a2390a9..19dca8a 100644
--- a/src/data.c
+++ b/src/data.c
@@ -44,10 +44,12 @@
char **random_paths = calloc(n, sizeof(char*));
int i;
pthread_mutex_lock(&mutex);
+ //printf("n = %d \n", n);
for(i = 0; i < n; ++i){
- int index = rand()%m;
+ int index = (rand()*rand())%m;
random_paths[i] = paths[index];
//if(i == 0) printf("%s\n", paths[index]);
+ //printf("%s\n", paths[index]);
}
pthread_mutex_unlock(&mutex);
return random_paths;
@@ -171,6 +173,13 @@
{
int i;
for(i = 0; i < n; ++i){
+ if(boxes[i].x == 0 && boxes[i].y == 0) {
+ boxes[i].x = 999999;
+ boxes[i].y = 999999;
+ boxes[i].w = 999999;
+ boxes[i].h = 999999;
+ continue;
+ }
boxes[i].left = boxes[i].left * sx - dx;
boxes[i].right = boxes[i].right * sx - dx;
boxes[i].top = boxes[i].top * sy - dy;
@@ -289,6 +298,7 @@
find_replace(path, "images", "labels", labelpath);
find_replace(labelpath, "JPEGImages", "labels", labelpath);
+ find_replace(labelpath, "raw", "labels", labelpath);
find_replace(labelpath, ".jpg", ".txt", labelpath);
find_replace(labelpath, ".png", ".txt", labelpath);
find_replace(labelpath, ".JPG", ".txt", labelpath);
@@ -309,7 +319,7 @@
h = boxes[i].h;
id = boxes[i].id;
- if (w < .01 || h < .01) continue;
+ if ((w < .01 || h < .01)) continue;
truth[i*5+0] = x;
truth[i*5+1] = y;
@@ -706,6 +716,7 @@
void *load_thread(void *ptr)
{
+ srand(time(0));
//printf("Loading data: %d\n", rand());
load_args a = *(struct load_args*)ptr;
if(a.exposure == 0) a.exposure = 1;
@@ -749,6 +760,7 @@
void *load_threads(void *ptr)
{
+ srand(time(0));
int i;
load_args args = *(load_args *)ptr;
if (args.threads == 0) args.threads = 1;
--
Gitblit v1.10.0