thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update yoloPlugins.h

Browse Source 
Corrected gridSize mistake in signature of YoloLayer class causing error at compile time.
This commit is contained in:
Pieris Kalligeros
2021-02-19 16:03:38 +02:00
committed by GitHub
parent 9dbb9edb7f
commit 59b9b9498a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
non_square/nvdsinfer_custom_impl_Yolo/yoloPlugins.h
View File

@@ -56,7 +56,7 @@ class YoloLayer : public nvinfer1::IPluginV2
{ {
public: public:
YoloLayer (const void* data, size_t length); 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 uint model_type, const uint new_coords, const float scale_x_y, const float beta_nms,
const std::vector<float> anchors, const std::vector<std::vector<int>> mask); const std::vector<float> anchors, const std::vector<std::vector<int>> mask);
const char* getPluginType () const override { return YOLOLAYER_PLUGIN_NAME; } const char* getPluginType () const override { return YOLOLAYER_PLUGIN_NAME; }