postprocessing.h

Go to the documentation of this file.

/*
 * Copyright (c) 2026 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
 */
 
#ifndef __POSTPROCESSING_H__
#define __POSTPROCESSING_H__
 
#include <stddef.h>
#include <stdint.h>
 
#include <drivers/axon/nrf_axon_nn_infer.h>
 
enum model_head {
        MODEL_HEAD_STRIDE_32, 
        MODEL_HEAD_STRIDE_16, 
        MODEL_HEAD_STRIDE_8,  
};
 
struct detection_box {
        float x1, y1, x2, y2;
        float score;
        enum model_head head_id;
};
 
void decode_init(const nrf_axon_nn_compiled_model_s *model);
 
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);
 
const char *model_head_name(enum model_head head_id);
 
#endif /* __POSTPROCESSING_H__ */