Fix ISliceLayer

2023-09-12 00:27:12 -03:00
parent d212d861cb
commit b2388fb3cb
9 changed files with 137 additions and 50 deletions
--- a/nvdsinfer_custom_impl_Yolo/layers/route_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/route_layer.h
@@ -8,7 +8,9 @@

 #include "../utils.h"

+#include "slice_layer.h"
+
 nvinfer1::ITensor* routeLayer(int layerIdx, std::string& layers, std::map<std::string, std::string>& block,
-    std::vector<nvinfer1::ITensor*> tensorOutputs, nvinfer1::INetworkDefinition* network);
+    std::vector<nvinfer1::ITensor*> tensorOutputs, nvinfer1::INetworkDefinition* network, uint batchSize);

 #endif