From 936d1480c57b94c73a072e7e636150ad46ff2b92 Mon Sep 17 00:00:00 2001
From: Marcos Luciano <marcoslucianops@gmail.com>
Date: Fri, 16 Jun 2023 12:18:22 -0300
Subject: [PATCH] Fix code

---
 nvdsinfer_custom_impl_Yolo/layers/channels_layer.h        | 1 +
 nvdsinfer_custom_impl_Yolo/layers/deconvolutional_layer.h | 1 +
 nvdsinfer_custom_impl_Yolo/layers/implicit_layer.h        | 1 +
 nvdsinfer_custom_impl_Yolo/layers/pooling_layer.h         | 1 +
 nvdsinfer_custom_impl_Yolo/layers/reorg_layer.h           | 1 +
 nvdsinfer_custom_impl_Yolo/layers/upsample_layer.h        | 1 +
 nvdsinfer_custom_impl_Yolo/utils.h                        | 1 +
 7 files changed, 7 insertions(+)

diff --git a/nvdsinfer_custom_impl_Yolo/layers/channels_layer.h b/nvdsinfer_custom_impl_Yolo/layers/channels_layer.h
index 6a2d389..6f580be 100644
--- a/nvdsinfer_custom_impl_Yolo/layers/channels_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/channels_layer.h
@@ -7,6 +7,7 @@
 #define __CHANNELS_LAYER_H__
 
 #include <map>
+#include <string>
 
 #include "NvInfer.h"
 
diff --git a/nvdsinfer_custom_impl_Yolo/layers/deconvolutional_layer.h b/nvdsinfer_custom_impl_Yolo/layers/deconvolutional_layer.h
index ae46e94..7e79b6c 100644
--- a/nvdsinfer_custom_impl_Yolo/layers/deconvolutional_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/deconvolutional_layer.h
@@ -8,6 +8,7 @@
 
 #include <map>
 #include <vector>
+#include <string>
 
 #include "NvInfer.h"
 
diff --git a/nvdsinfer_custom_impl_Yolo/layers/implicit_layer.h b/nvdsinfer_custom_impl_Yolo/layers/implicit_layer.h
index 10e50e2..dd5462a 100644
--- a/nvdsinfer_custom_impl_Yolo/layers/implicit_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/implicit_layer.h
@@ -8,6 +8,7 @@
 
 #include <map>
 #include <vector>
+#include <string>
 
 #include "NvInfer.h"
 
diff --git a/nvdsinfer_custom_impl_Yolo/layers/pooling_layer.h b/nvdsinfer_custom_impl_Yolo/layers/pooling_layer.h
index 82f1971..f4d0168 100644
--- a/nvdsinfer_custom_impl_Yolo/layers/pooling_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/pooling_layer.h
@@ -7,6 +7,7 @@
 #define __POOLING_LAYER_H__
 
 #include <map>
+#include <string>
 
 #include "NvInfer.h"
 
diff --git a/nvdsinfer_custom_impl_Yolo/layers/reorg_layer.h b/nvdsinfer_custom_impl_Yolo/layers/reorg_layer.h
index 585b91d..e2f0866 100644
--- a/nvdsinfer_custom_impl_Yolo/layers/reorg_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/reorg_layer.h
@@ -7,6 +7,7 @@
 #define __REORG_LAYER_H__
 
 #include <map>
+#include <string>
 
 #include "NvInfer.h"
 
diff --git a/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.h b/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.h
index 546db2f..66f9702 100644
--- a/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.h
@@ -7,6 +7,7 @@
 #define __UPSAMPLE_LAYER_H__
 
 #include <map>
+#include <string>
 
 #include "NvInfer.h"
 
diff --git a/nvdsinfer_custom_impl_Yolo/utils.h b/nvdsinfer_custom_impl_Yolo/utils.h
index af7e202..4de00de 100644
--- a/nvdsinfer_custom_impl_Yolo/utils.h
+++ b/nvdsinfer_custom_impl_Yolo/utils.h
@@ -28,6 +28,7 @@
 
 #include <map>
 #include <vector>
+#include <string>
 #include <cassert>
 #include <iostream>
 #include <fstream>