5 RT-AK定义的输入输出与模型不匹配？

发布于 2021-09-13 10:57:34 浏览：657 订阅该版

我发现模型是float输入，float输出，但看到RT-AK定义的输入输出是u8格式？这怎么修改呢？
```c
struct rt_ai
{
    struct rt_ai_core   parent;
    rt_ai_uint16_t      flag; /**< ai flag */
    rt_ai_uint16_t      mem_flag;
    rt_ai_buffer_t      *workbuffer;           /** runtime buf */
    rt_ai_buffer_t      *input[RT_AI_IO_MAX];  /* num of input */
    rt_ai_buffer_t      *output[RT_AI_IO_MAX]; /* num of output */
    struct rt_ai_info   info;
    void                *cfg; /* platform config */

#ifdef RT_USING_AI_OPS
    const struct rt_ai_ops *ops;
#else
    /* common ai interface 
     *@return 0:OK; other:fail
    */
    int (*init)         (rt_ai_t ai, rt_ai_buffer_t *buf);
    int (*get_input)    (rt_ai_t ai, rt_ai_uint32_t index);
    int (*run)          (rt_ai_t ai, void (*callback)(void *arg), void *arg);
    int (*get_output)   (rt_ai_t ai, rt_ai_uint32_t index);
    int (*get_info)     (rt_ai_t ai, rt_ai_buffer_t *buf);
    int (*config)       (rt_ai_t ai, int cmd, rt_ai_buffer_t *args);

#endif
    /* callback when run done */
    void (*done_callback_user)(void *arg);
    void *arg;
};
```

RT-AK定义的rt_ai_buffer_t
```c
typedef rt_uint8_t  rt_ai_buffer_t;
```

想请教如何修改keras输入输出格式，或者不修改模式修改代码保留float输入输出的功能？