StageRun.h File

Included Headers

#include "pipeline/DetectionTypes.h" #include "pipeline/TensorCore.h" #include <string>

Namespaces Index

namespace	cv
namespace	simaai
namespace	neat
namespace	stages

Classes Index

struct	BoxDecodeOptions

File Listing

The file content with the documentation metadata removed is:

1#pragma once

2

3#include "pipeline/DetectionTypes.h"

4#include "pipeline/TensorCore.h"

5

6#include <string>

7

8namespace cv {

9class Mat;

10} // namespace cv

11

12namespace simaai::neat {

13class Model;

14

15namespace stages {

16

17struct BoxDecodeOptions {

18 std::string decode_type = "";

19 int original_width = 0;

20 int original_height = 0;

21 double detection_threshold = 0.0;

22 double nms_iou_threshold = 0.0;

23 int top_k = 0;

24};

25

26simaai::neat::Tensor Preproc(const cv::Mat& input, const simaai::neat::Model& model);

27simaai::neat::Tensor Infer(const simaai::neat::Tensor& input, const simaai::neat::Model& model);

28simaai::neat::Tensor MLA(const simaai::neat::Tensor& input, const simaai::neat::Model& model);

29BoxDecodeResult BoxDecode(const simaai::neat::Tensor& input, const simaai::neat::Model& model,

30 const BoxDecodeOptions& opt = {});

31

32} // namespace stages

33} // namespace simaai::neat

---

Generated via doxygen2docusaurus 2.0.0 by Doxygen 1.9.1.