You're reading an pre-release version of this documentation.
For the latest stable release version, please have a look at master.

Dismiss

Program Listing for File preprocess.h

↰ Return to documentation for file (include/preprocess.h)

#ifndef _PREPROCESS_H_
#define _PREPROCESS_H_

#ifdef __cplusplus
extern "C" {
#endif

#include "quantize.h"

typedef struct _normalization_params
{
    int mean[3];
    float scale;
} normalization_params_t;


void set_normalize_params(normalization_params_t *norm_params, \
                          int *mean, float scale);


float normalize_u8(uint8_t input, int mean, float scale);


int normalize_quantize(buffer_t *output, buffer_t *input,
                       normalization_params_t *norm_params,
                       pnna_buffer_create_params_t *quant_params,
                       uint32_t counts);


int quantize(buffer_t *output, buffer_t *input,
             pnna_buffer_create_params_t *quant_params,
             uint32_t counts);


buffer_t *create_buffer_from_qparams(pnna_buffer_create_params_t *qp, uint32_t count);


void destroy_buffer_array(buffer_t *buff_array, uint32_t count);


#ifdef __linux__
buffer_t load_binary_to_buffer(const char *name);


void free_buffer(buffer_t *buffer);


buffer_t load_image(const char *filename);


buffer_t load_image_from_memory(const uint8_t *buffer, int len);


void free_image_buffer(buffer_t *buffer);

#endif


#ifdef __cplusplus
}
#endif

#endif // !_PREPROCESS_H_