From 8b9af80a5b3bdb6ec17c4fa2cfa14ce4676ef1c3 Mon Sep 17 00:00:00 2001
From: Alexey <AlexeyAB@users.noreply.github.com>
Date: Fri, 18 May 2018 20:27:59 +0000
Subject: [PATCH] Merge pull request #851 from NadeenUdantha/patch-1

---
 src/darknet.c |    9 +++++++++
 1 files changed, 9 insertions(+), 0 deletions(-)

diff --git a/src/darknet.c b/src/darknet.c
index 0f6af48..5407269 100644
--- a/src/darknet.c
+++ b/src/darknet.c
@@ -1,3 +1,8 @@
+#ifdef _DEBUG
+#include <stdlib.h> 
+#include <crtdbg.h>  
+#endif
+
 #include <time.h>
 #include <stdlib.h>
 #include <stdio.h>
@@ -350,6 +355,10 @@
 
 int main(int argc, char **argv)
 {
+#ifdef _DEBUG
+	_CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF);
+#endif
+
     //test_resize("data/bad.jpg");
     //test_box();
     //test_convolutional_layer();

--
Gitblit v1.10.0