Yolo3Detection.cpp doesn't build yet,issues with dependecies of preprocessing and postprocessing on IPluginFactory
This commit is contained in:
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user