multi-task-v2(instance+semantic)

src/libs/postprocessors/multi_post_v2.h

+
+#ifndef DEEPINFER_POSTPROCESS_MULTI_H_
+#define DEEPINFER_POSTPROCESS_MULTI_H_
+
+#include "interfaces/base_unit.h"
+#include "base/image.h"
+
+#include "libs/ios/normal_ios.h"
+#include "libs/ios/camera_ios.h"
+#include "libs/ios/detection.h"
+#include "libs/ios/semantic.h"
+#include "libs/ios/depth.h"
+#include "libs/postprocessors/multi_post_gpu.h"
+#include "libs/postprocessors/nms.h"
+
+namespace waytous {
+namespace deepinfer {
+namespace postprocess {
+
+
+class MultiPostProcessV2: public interfaces::BaseUnit{
+
+public:
+    bool Init(YAML::Node& node, interfaces::BaseIOMapPtr pmap) override;
+
+    bool Exec() override;
+
+    std::string Name() override;
+
+
+public:
+    base::BlobPtr<int> output_length_ptr;
+    base::BlobPtr<float> bboxes_ptr;
+
+    int inputHeight, inputWidth;
+    int inferBatchSize = 1;
+    bool fixAspectRatio = true;
+    float nmsThreshold, scoreThreshold, truncatedThreshold;
+    int rawDetectionLength;
+    int maxOutputNum = 1000;
+    int keepTopK = 100;
+    int segProtoDim = 32;
+    int instanceDownScale = 4;
+    int instanceClassNumber;
+    std::vector<std::string> instanceClassNames;
+    int detectionStep;
+
+    int semanticDownScale = 4, semanticClassNumber;
+    std::vector<std::string> semanticClassNames;
+
+    // int depthDownScale = 4, depthDistanceScale = 256;
+};
+
+DEEPINFER_REGISTER_UNIT(MultiPostProcessV2);
+
+
+}  // namespace postprocess
+}  // namespace deepinfer
+}  // namespace waytous
+
+#endif
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+