mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-02-24 08:23:52 +08:00
160c9ba1d2
* Navmodel and driving style: update runner * No driving style yet * thneedrunner without extra inputs * nav feature ref
36 lines
996 B
C++
36 lines
996 B
C++
#pragma once
|
|
|
|
#include "selfdrive/modeld/runners/runmodel.h"
|
|
#include "selfdrive/modeld/thneed/thneed.h"
|
|
|
|
class ThneedModel : public RunModel {
|
|
public:
|
|
ThneedModel(const char *path, float *loutput, size_t loutput_size, int runtime, bool luse_extra = false, bool use_tf8 = false, cl_context context = NULL);
|
|
void addRecurrent(float *state, int state_size);
|
|
void addTrafficConvention(float *state, int state_size);
|
|
void addDesire(float *state, int state_size);
|
|
void addNavFeatures(float *state, int state_size);
|
|
void addDrivingStyle(float *state, int state_size);
|
|
void addImage(float *image_buf, int buf_size);
|
|
void addExtra(float *image_buf, int buf_size);
|
|
void execute();
|
|
void* getInputBuf();
|
|
void* getExtraBuf();
|
|
private:
|
|
Thneed *thneed = NULL;
|
|
bool recorded;
|
|
bool use_extra;
|
|
|
|
float *input;
|
|
float *extra;
|
|
float *output;
|
|
|
|
// recurrent and desire
|
|
float *recurrent;
|
|
float *trafficConvention;
|
|
float *drivingStyle;
|
|
float *desire;
|
|
float *navFeatures;
|
|
};
|
|
|