DeepStream 7.1 + Fixes + New model output format

2024-11-07 11:25:17 -03:00
parent bca9e59d07
commit b451b036b2
75 changed files with 2383 additions and 1113 deletions
--- a/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp
+++ b/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp
@@ -24,7 +24,13 @@ upsampleLayer(int layerIdx, std::map<std::string, std::string>& block, nvinfer1:
  assert(resize != nullptr);
  std::string resizeLayerName = "upsample_" + std::to_string(layerIdx);
  resize->setName(resizeLayerName.c_str());
+  
+#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 0
+  resize->setResizeMode(nvinfer1::InterpolationMode::kNEAREST);
+#else
  resize->setResizeMode(nvinfer1::ResizeMode::kNEAREST);
+#endif
+
  resize->setScales(scale, 4);
  output = resize->getOutput(0);