Rebase Python example from ioir123ju/tkDNN/master/

include/tkDNN/DetectionNN.h

@@ -138,7 +138,19 @@ class DetectionNN {
                 TKDNN_TSTOP
                 if(save_times) *times<<t_ns<<"\n";
             }
-        }      
+        }
+        /**
+         * Method to get results.
+         *
+         */
+        std::vector<std::vector<tk::dnn::box>>& get_batch_detected()
+        {
+            return batchDetected;
+        }
+        std::vector<std::string> get_classesName()
+        {
+            return classesNames;
+        }
 
         /**
          * Method to draw boundixg boxes and labels on a frame.

include/tkDNN/utils.h

@@ -6,7 +6,7 @@
 #include <fstream>
 #include <iomanip>
 #include <stdlib.h>
-
+#include <malloc.h>
 #include "cuda.h"
 #include "cuda_runtime_api.h"
 #include <cublas_v2.h>
@@ -101,6 +101,12 @@ typedef enum {
   ERROR_CUDNNvsTENSORRT = 8    
 } resultError_t;
 
+void *xmalloc(size_t size);
+void *xcalloc(size_t nmemb, size_t size);
+void malloc_error();
+void calloc_error();
+void realloc_error();
+
 void printCenteredTitle(const char *title, char fill, int dim = 30);
 bool fileExist(const char *fname);
 void downloadWeightsifDoNotExist(const std::string& input_bin, const std::string& test_folder, const std::string& weights_url);