mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-02-21 10:03:53 +08:00
85efde269d
* Added wide cam vipc client and bigmodel transform logic * Added wide_frame to ModelState, should still work normally * Refactored image input into addImage method, should still work normally * Updated thneed/compile.cc * Bigmodel, untested: 44f83118-b375-4d4c-ae12-2017124f0cf4/200 * Have to initialize extra buffer in SNPEModel * Default paramater value in the wrong place I think * Move USE_EXTRA to SConscript * New model: 6c34d59a-acc3-4877-84bd-904c10745ba6/250 * move use extra check to runtime, not on C2 * this is always true * more C2 checks * log if frames are out of sync * more logging on no frame * store in pointer * print sof * add sync logic * log based on sof difference as well * keep both models * less assumptions * define above thneed * typo * simplify * no need for second client is main is already wide * more comments update * no optional reference * more logging to debug lags * add to release files * both defines * New model: 6831a77f-2574-4bfb-8077-79b0972a2771/950 * Path offset no longer relevant * Remove duplicate execute * Moved bigmodel back to big_supercombo.dlc * add wide vipc stream * Tici must be tici * Needs state too * add wide cam support to model replay * handle syncing better * ugh, c2 * print that * handle ecam lag * skip first one * so close * update refs Co-authored-by: mitchellgoffpc <mitchellgoffpc@gmail.com> Co-authored-by: Harald Schafer <harald.the.engineer@gmail.com> Co-authored-by: Adeeb Shihadeh <adeebshihadeh@gmail.com> Co-authored-by: Comma Device <device@comma.ai>
32 lines
795 B
C++
32 lines
795 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);
|
|
void addRecurrent(float *state, int state_size);
|
|
void addTrafficConvention(float *state, int state_size);
|
|
void addDesire(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 *desire;
|
|
};
|
|
|