dm e2e outputs added (#21210)

* try this

* use e2e only

* add new outputs

* Revert "use e2e only"

This reverts commit 423dd8e29b.

* send new

* fix batchnorm
pull/21214/head
ZwX1616 2021-06-09 16:18:14 -07:00 committed by GitHub
parent a361441fe2
commit 46e3f4528f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 13 additions and 7 deletions

2
cereal

@ -1 +1 @@
Subproject commit cde8667d3bacb71d324b06b3d803c8fd7c59d2db
Subproject commit 018e7b5c9bd1ea65baf98326f8c95f849551d5c0

View File

@ -1,2 +1,2 @@
6272770f-5230-4a10-ae5e-c8f3d205ea96
0087a1181516dd0c64d7f7a58d53905e2cc76b72
abbc836aaa1e940c56673b6f42f5762ad71b35e1

BIN
models/dmonitoring_model.onnx (Stored with Git LFS)

Binary file not shown.

BIN
models/dmonitoring_model_q.dlc (Stored with Git LFS)

Binary file not shown.

View File

@ -162,6 +162,8 @@ DMonitoringResult dmonitoring_eval_frame(DMonitoringModelState* s, void* stream_
ret.partial_face = s->output[35];
ret.distracted_pose = s->output[36];
ret.distracted_eyes = s->output[37];
ret.eyes_on_road = s->output[38];
ret.phone_use = s->output[39];
ret.dsp_execution_time = (t2 - t1) / 1000.;
return ret;
}
@ -188,6 +190,8 @@ void dmonitoring_publish(PubMaster &pm, uint32_t frame_id, const DMonitoringResu
framed.setPartialFace(res.partial_face);
framed.setDistractedPose(res.distracted_pose);
framed.setDistractedEyes(res.distracted_eyes);
framed.setEyesOnRoad(res.eyes_on_road);
framed.setPhoneUse(res.phone_use);
if (send_raw_pred) {
framed.setRawPredictions(raw_pred.asBytes());
}

View File

@ -7,7 +7,7 @@
#include "selfdrive/modeld/models/commonmodel.h"
#include "selfdrive/modeld/runners/run.h"
#define OUTPUT_SIZE 38
#define OUTPUT_SIZE 40
typedef struct DMonitoringResult {
float face_orientation[3];
@ -24,6 +24,8 @@ typedef struct DMonitoringResult {
float partial_face;
float distracted_pose;
float distracted_eyes;
float eyes_on_road;
float phone_use;
float dsp_execution_time;
} DMonitoringResult;