Yolo3Detection.cpp doesn't build yet,issues with dependecies of preprocessing and postprocessing on IPluginFactory

This commit is contained in:
perseusdg
2021-08-21 22:29:36 +05:30
parent eba78e7e78
commit 3d8b1ac494
7 changed files with 65 additions and 60 deletions
+4
View File
@@ -2,6 +2,7 @@
#include "../kernels.h"
#define YOLORT_CLASSNAME_W 256
class YoloRT : public IPluginV2 {
public:
@@ -49,6 +50,7 @@ public:
classesNames[1] = std::string(tmp);
}
assert(buf == bufCheck + length);
}
~YoloRT() {
@@ -56,6 +58,7 @@ public:
}
int getNbOutputs() const NOEXCEPT override {
return 1;
}
@@ -186,6 +189,7 @@ public:
float nms_thresh;
int nms_kind;
int new_coords;
int NUM=0;
std::vector<std::string> classesNames;
dnnType *mask;