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

Merge pull request #35 from pieris98/master

Browse Source 
Update yoloPlugins.h
This commit is contained in:
Marcos Luciano
2021-02-19 11:51:18 -03:00
committed by GitHub
parent 9dbb9edb7f 59b9b9498a
commit 3f3ccbeb73
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
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; }
@@ -153,4 +153,4 @@ extern float kBETA_NMS;
extern std::vector<float> kANCHORS; extern std::vector<float> kANCHORS;
extern std::vector<std::vector<int>> kMASK; extern std::vector<std::vector<int>> kMASK;
#endif // __YOLO_PLUGINS__ #endif // __YOLO_PLUGINS__