From 59b9b9498a0705797797d6373e2a7a7165ff49da Mon Sep 17 00:00:00 2001 From: Pieris Kalligeros <34139689+pieris98@users.noreply.github.com> Date: Fri, 19 Feb 2021 16:03:38 +0200 Subject: [PATCH] Update yoloPlugins.h Corrected gridSize mistake in signature of YoloLayer class causing error at compile time. --- non_square/nvdsinfer_custom_impl_Yolo/yoloPlugins.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/non_square/nvdsinfer_custom_impl_Yolo/yoloPlugins.h b/non_square/nvdsinfer_custom_impl_Yolo/yoloPlugins.h index 2981af7..177ca10 100644 --- a/non_square/nvdsinfer_custom_impl_Yolo/yoloPlugins.h +++ b/non_square/nvdsinfer_custom_impl_Yolo/yoloPlugins.h @@ -56,7 +56,7 @@ class YoloLayer : public nvinfer1::IPluginV2 { public: YoloLayer (const void* data, size_t length); - YoloLayer (const uint& numBoxes, const uint& numClasses, const uint& gridSize, + YoloLayer (const uint& numBoxes, const uint& numClasses, const uint& gridSizeX, const uint& gridSizeY, const uint model_type, const uint new_coords, const float scale_x_y, const float beta_nms, const std::vector anchors, const std::vector> mask); const char* getPluginType () const override { return YOLOLAYER_PLUGIN_NAME; } @@ -153,4 +153,4 @@ extern float kBETA_NMS; extern std::vector kANCHORS; extern std::vector> kMASK; -#endif // __YOLO_PLUGINS__ \ No newline at end of file +#endif // __YOLO_PLUGINS__