From 35a66b73a25ab93f2a3ceb85993d4801240046a4 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Thu, 01 Mar 2018 12:14:58 +0000
Subject: [PATCH] Changed class to class_id
---
src/box.c | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/box.c b/src/box.c
index 9568599..1dc12c0 100644
--- a/src/box.c
+++ b/src/box.c
@@ -232,7 +232,7 @@
typedef struct{
int index;
- int class;
+ int class_id;
float **probs;
} sortable_bbox;
@@ -240,7 +240,7 @@
{
sortable_bbox a = *(sortable_bbox *)pa;
sortable_bbox b = *(sortable_bbox *)pb;
- float diff = a.probs[a.index][b.class] - b.probs[b.index][b.class];
+ float diff = a.probs[a.index][b.class_id] - b.probs[b.index][b.class_id];
if(diff < 0) return 1;
else if(diff > 0) return -1;
return 0;
@@ -253,13 +253,13 @@
for(i = 0; i < total; ++i){
s[i].index = i;
- s[i].class = 0;
+ s[i].class_id = 0;
s[i].probs = probs;
}
for(k = 0; k < classes; ++k){
for(i = 0; i < total; ++i){
- s[i].class = k;
+ s[i].class_id = k;
}
qsort(s, total, sizeof(sortable_bbox), nms_comparator);
for(i = 0; i < total; ++i){
--
Gitblit v1.10.0