thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix code

Browse Source
This commit is contained in:
Marcos Luciano
2023-06-16 12:18:22 -03:00
parent 64fa573f72
commit 936d1480c5
7 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
nvdsinfer_custom_impl_Yolo/layers/channels_layer.h
View File

@@ -7,6 +7,7 @@
#define __CHANNELS_LAYER_H__ #define __CHANNELS_LAYER_H__
#include <map> #include <map>
#include <string>
#include "NvInfer.h" #include "NvInfer.h"

1
nvdsinfer_custom_impl_Yolo/layers/deconvolutional_layer.h
View File

@@ -8,6 +8,7 @@
#include <map> #include <map>
#include <vector> #include <vector>
#include <string>
#include "NvInfer.h" #include "NvInfer.h"

1
nvdsinfer_custom_impl_Yolo/layers/implicit_layer.h
View File

@@ -8,6 +8,7 @@
#include <map> #include <map>
#include <vector> #include <vector>
#include <string>
#include "NvInfer.h" #include "NvInfer.h"

1
nvdsinfer_custom_impl_Yolo/layers/pooling_layer.h
View File

@@ -7,6 +7,7 @@
#define __POOLING_LAYER_H__ #define __POOLING_LAYER_H__
#include <map> #include <map>
#include <string>
#include "NvInfer.h" #include "NvInfer.h"

1
nvdsinfer_custom_impl_Yolo/layers/reorg_layer.h
View File

@@ -7,6 +7,7 @@
#define __REORG_LAYER_H__ #define __REORG_LAYER_H__
#include <map> #include <map>
#include <string>
#include "NvInfer.h" #include "NvInfer.h"

1
nvdsinfer_custom_impl_Yolo/layers/upsample_layer.h
View File

@@ -7,6 +7,7 @@
#define __UPSAMPLE_LAYER_H__ #define __UPSAMPLE_LAYER_H__
#include <map> #include <map>
#include <string>
#include "NvInfer.h" #include "NvInfer.h"

1
nvdsinfer_custom_impl_Yolo/utils.h
View File

@@ -28,6 +28,7 @@
#include <map> #include <map>
#include <vector> #include <vector>
#include <string>
#include <cassert> #include <cassert>
#include <iostream> #include <iostream>
#include <fstream> #include <fstream>