Edge AI Add-on API 2.2.0
Loading...
Searching...
No Matches
postprocessing.h File Reference
#include <stddef.h>
#include <stdint.h>
#include <drivers/axon/nrf_axon_nn_infer.h>

Go to the source code of this file.

Data Structures

struct  detection_box
 Bounding box for object detection. More...
 

Enumerations

enum  model_head { MODEL_HEAD_STRIDE_32 , MODEL_HEAD_STRIDE_16 , MODEL_HEAD_STRIDE_8 }
 Model head ID. More...
 

Functions

void decode_init (const nrf_axon_nn_compiled_model_s *model)
 Initialize decode procedure.
 
size_t decode_output (const nrf_axon_nn_compiled_model_s *model, const int8_t *packed_data, struct detection_box *boxes, const size_t boxes_size)
 Decode output from the model and perform NMS.
 
const char * model_head_name (enum model_head head_id)
 Gives string name of model's head.