Big update

2023-05-19 03:05:43 -03:00
parent 68f762d5bd
commit 07feae9509
86 changed files with 1523 additions and 5223 deletions
--- a/nvdsinfer_custom_impl_Yolo/layers/reorg_layer.cpp
+++ b/nvdsinfer_custom_impl_Yolo/layers/reorg_layer.cpp
@@ -14,7 +14,7 @@ reorgLayer(int layerIdx, std::map<std::string, std::string>& block, nvinfer1::IT
 {
  nvinfer1::ITensor* output;

-  assert(block.at("type") == "reorg");
+  assert(block.at("type") == "reorg3d");

  nvinfer1::Dims inputDims = input->getDimensions();