From ec68838342b42541776607b0c14e40fb89f7e3d8 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 23 May 2018 15:27:18 +0000
Subject: [PATCH] Fixed memory leaks for Yolo: train, test

---
 src/box.h |    5 +++++
 1 files changed, 5 insertions(+), 0 deletions(-)

diff --git a/src/box.h b/src/box.h
index 6063a48..9553b12 100644
--- a/src/box.h
+++ b/src/box.h
@@ -1,6 +1,11 @@
 #ifndef BOX_H
 #define BOX_H
 
+#ifdef _DEBUG
+#include <stdlib.h> 
+#include <crtdbg.h>  
+#endif
+
 #ifdef YOLODLL_EXPORTS
 #if defined(_MSC_VER)
 #define YOLODLL_API __declspec(dllexport) 

--
Gitblit v1.10.0