inference_postprocessing.h

Go to the documentation of this file.

/*
 * Copyright (c) 2026 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
 */
 
#ifndef __INFERENCE_POSTPROCESSING_H__
#define __INFERENCE_POSTPROCESSING_H__
 
#include <stdint.h>
 
typedef enum {
        CLASS_LABEL_IDLE,
        CLASS_LABEL_UNKNOWN,
        CLASS_LABEL_SWIPE_RIGHT,
        CLASS_LABEL_SWIPE_LEFT,
        CLASS_LABEL_DOUBLE_SHAKE,
        CLASS_LABEL_DOUBLE_THUMB,
        CLASS_LABEL_ROTATION_RIGHT,
        CLASS_LABEL_ROTATION_LEFT,
        CLASS_LABEL_COUNT,
} class_label_t;
 
typedef struct prediction_ctx_s {
        uint16_t target;
 
        float probability;
} prediction_ctx_t;
 
prediction_ctx_t inference_postprocess(const uint16_t predicted_target,
                                       const float probability);
 
const char *inference_get_class_name(const class_label_t class_label);
 
 
 
#endif /* __INFERENCE_POSTPROCESSING_H__ */