From 2662cd51a32cf8292ccce555f64880f460108d2d Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Tue, 28 Mar 2017 10:57:27 +0000
Subject: [PATCH] Fixed sequence of freeing memory.

---
 src/yolo_v2_class.cpp |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/src/yolo_v2_class.cpp b/src/yolo_v2_class.cpp
index fb06985..ea13ea3 100644
--- a/src/yolo_v2_class.cpp
+++ b/src/yolo_v2_class.cpp
@@ -79,9 +79,9 @@
 	for (int j = 0; j < FRAMES; ++j) free(detector_gpu.predictions[j]);
 	for (int j = 0; j < FRAMES; ++j) if(detector_gpu.images[j].data) free(detector_gpu.images[j].data);
 
+	for (int j = 0; j < l.w*l.h*l.n; ++j) free(detector_gpu.probs[j]);
 	free(detector_gpu.boxes);
 	free(detector_gpu.probs);
-	for (int j = 0; j < l.w*l.h*l.n; ++j) free(detector_gpu.probs[j]);
 
 	int old_gpu_index;
 	cudaGetDevice(&old_gpu_index);

--
Gitblit v1.10.0