From 5995048f6348a44d6a3eb9d64ece9d402a6d8899 Mon Sep 17 00:00:00 2001
From: hchandirasekar <f20180523@goa.bits-pilani.ac.in>
Date: Mon, 25 Jan 2021 04:12:05 +0530
Subject: [PATCH] Replaced dynamic arrays with std::vector ,works on linux
 ..needs to be tested on windows after clearing up the lnk2019 error

---
 CMakeLists.txt         | 4 ++--
 src/Yolo3Detection.cpp | 7 ++++---
 src/utils.cpp          | 2 +-
 3 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 03e26c5..c11b56f 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -2,13 +2,13 @@ cmake_minimum_required(VERSION 3.5)
 
 project (tkDNN)
 set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
-if(LINUX)
+if(UNIX)
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -fPIC -Wno-deprecated-declarations -Wno-unused-variable")
 endif()
 if(WIN32)
 set(CMAKE_CXX_STANDARD 14)
 set(CMAKE_CXX_FLAGS "/O2 /FS ")
-endif()
+endif(WIN32)
 include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include/tkDNN)
 
 # project specific flags
diff --git a/src/Yolo3Detection.cpp b/src/Yolo3Detection.cpp
index b94eea9..0c638e6 100644
--- a/src/Yolo3Detection.cpp
+++ b/src/Yolo3Detection.cpp
@@ -94,9 +94,10 @@ void Yolo3Detection::preprocess(cv::Mat &frame, const int bi){
 void Yolo3Detection::postprocess(const int bi, const bool mAP){
 
     //get yolo outputs
-    dnnType *rt_out[netRT->pluginFactory->n_yolos]; 
-    for(int i=0; i<netRT->pluginFactory->n_yolos; i++) 
-        rt_out[i] = (dnnType*)netRT->buffersRT[i+1] + netRT->buffersDIM[i+1].tot()*bi;
+    std::vector<float *> rt_out;
+    //dnnType *rt_out[netRT->pluginFactory->n_yolos];
+    for(int i=0; i<netRT->pluginFactory->n_yolos; i++)
+        rt_out.push_back((dnnType*)netRT->buffersRT[i+1] + netRT->buffersDIM[i+1].tot()*bi);
 
     float x_ratio =  float(originalSize[bi].width) / float(netRT->input_dim.w);
     float y_ratio =  float(originalSize[bi].height) / float(netRT->input_dim.h);
diff --git a/src/utils.cpp b/src/utils.cpp
index e143cce..bbdf516 100644
--- a/src/utils.cpp
+++ b/src/utils.cpp
@@ -170,7 +170,6 @@ void getMemUsage(double& vm_usage_kb, double& resident_set_kb){
    using std::ios_base;
    using std::ifstream;
    using std::string;
-   SYSTEM_INFO sysInfo;
 
    vm_usage_kb     = 0.0;
    resident_set_kb = 0.0;
@@ -195,6 +194,7 @@ void getMemUsage(double& vm_usage_kb, double& resident_set_kb){
 #ifdef __linux__
    long page_size_kb = sysconf(_SC_PAGE_SIZE) / 1024; // in case x86-64 is configured to use 2MB pages
 #elif _WIN32
+SYSTEM_INFO sysInfo;
  long page_size_kb = sysInfo.dwPageSize/1024;
 #endif