Skip to main content

DetectionTypes.h File

Detection helpers for decoding model outputs. More...

Included Headers​

#include "pipeline/TensorCore.h" #include <cstdint> #include <string> #include <vector>

Namespaces Index​

namespacesimaai
namespaceneat

Classes Index​

structBox
structBoxDecodeResult

Description​

Detection helpers for decoding model outputs.

File Listing​

The file content with the documentation metadata removed is:

1
6#pragma once
7
8#include "pipeline/TensorCore.h"
9
10#include <cstdint>
11#include <string>
12#include <vector>
13
14namespace simaai::neat {
15
16struct Box {
17 float x1 = 0.0f;
18 float y1 = 0.0f;
19 float x2 = 0.0f;
20 float y2 = 0.0f;
21 float score = 0.0f;
22 int class_id = -1;
23};
24
25struct BoxDecodeResult {
26 std::vector<Box> boxes;
27 std::vector<uint8_t> raw;
28};
29
30std::vector<Box> parse_bbox_bytes(const std::vector<uint8_t>& bytes, int img_w, int img_h,
31 int expected_topk, bool strict);
32
33BoxDecodeResult decode_bbox_tensor(const simaai::neat::Tensor& tensor, int img_w, int img_h,
34 int expected_topk, bool strict);
35
36} // namespace simaai::neat

Generated via doxygen2docusaurus 2.0.0 by Doxygen 1.9.1.