From 59e356730fa62d1e307d4fc7a5657f8492633088 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 22 Sep 2015 22:24:54 +0000
Subject: [PATCH] writing stuff
---
src/swag.c | 19 ++++++++++---------
1 files changed, 10 insertions(+), 9 deletions(-)
diff --git a/src/swag.c b/src/swag.c
index 1398192..37dde36 100644
--- a/src/swag.c
+++ b/src/swag.c
@@ -132,21 +132,22 @@
void convert_swag_detections(float *predictions, int classes, int num, int square, int side, int w, int h, float thresh, float **probs, box *boxes)
{
int i,j,n;
- int per_cell = 5*num+classes;
+ //int per_cell = 5*num+classes;
for (i = 0; i < side*side; ++i){
int row = i / side;
int col = i % side;
for(n = 0; n < num; ++n){
- int offset = i*per_cell + 5*n;
- float scale = predictions[offset];
int index = i*num + n;
- boxes[index].x = (predictions[offset + 1] + col) / side * w;
- boxes[index].y = (predictions[offset + 2] + row) / side * h;
- boxes[index].w = pow(predictions[offset + 3], (square?2:1)) * w;
- boxes[index].h = pow(predictions[offset + 4], (square?2:1)) * h;
+ int p_index = side*side*classes + i*num + n;
+ float scale = predictions[p_index];
+ int box_index = side*side*(classes + num) + (i*num + n)*4;
+ boxes[index].x = (predictions[box_index + 0] + col) / side * w;
+ boxes[index].y = (predictions[box_index + 1] + row) / side * h;
+ boxes[index].w = pow(predictions[box_index + 2], (square?2:1)) * w;
+ boxes[index].h = pow(predictions[box_index + 3], (square?2:1)) * h;
for(j = 0; j < classes; ++j){
- offset = i*per_cell + 5*num;
- float prob = scale*predictions[offset+j];
+ int class_index = i*classes;
+ float prob = scale*predictions[class_index+j];
probs[index][j] = (prob > thresh) ? prob : 0;
}
}
--
Gitblit v1.10.0