model_utils.h

Go to the documentation of this file.

/*
 * Copyright (c) 2026 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
 */
 
#ifndef MODEL_UTILS_H_
#define MODEL_UTILS_H_
 
#include <errno.h>
#include <stdlib.h>
 
#include <zephyr/sys/__assert.h>
 
#include <nrf_edgeai/nrf_edgeai.h>
#include <nrf_edgeai_obsv/nrf_edgeai_obsv_core.h>
 
static inline int obsv_model_info_from_model(nrf_edgeai_t *model, uint16_t num_classes,
                                             nrf_edgeai_obsv_model_info_t *info)
{
        __ASSERT_NO_MSG(model != NULL);
        __ASSERT_NO_MSG(info != NULL);
        __ASSERT_NO_MSG(nrf_edgeai_model_outputs_num(model) == num_classes);
 
        const unsigned long model_id = strtoul(model->metadata.p_solution_id, NULL, 10);
 
        if (model_id > UINT16_MAX) {
                return -EINVAL;
        }
 
        const nrf_edgeai_dsp_feature_extraction_t *features_ctx =
                nrf_edgeai_dsp_features_ctx(model);
 
        info->model_id = (uint16_t)model_id;
        info->num_classes = num_classes;
        info->num_features = features_ctx != NULL ? features_ctx->overall_num : 0;
        info->version = model->metadata.version.combined;
 
        return 0;
}
 
#endif /* MODEL_UTILS_H_ */