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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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