model replay: add cameraOdometry (#30007)

* model replay: add back cameraOdometry coverage

* update refs

* try again

* fix cmp log
pull/30019/head
Adeeb Shihadeh 2 years ago committed by GitHub
parent e16d530da9
commit 5f318faf93
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 28
      selfdrive/test/process_replay/model_replay.py
  2. 2
      selfdrive/test/process_replay/model_replay_ref_commit

@ -7,7 +7,6 @@ from typing import Any
import cereal.messaging as messaging
from openpilot.common.params import Params
from openpilot.common.spinner import Spinner
from openpilot.system.hardware import PC
from openpilot.selfdrive.manager.process_config import managed_processes
from openpilot.selfdrive.test.openpilotci import BASE_URL, get_url
@ -105,14 +104,6 @@ def nav_model_replay(lr):
def model_replay(lr, frs):
if not PC:
spinner = Spinner()
spinner.update("starting model replay")
else:
spinner = None
log_msgs = []
# modeld is using frame pairs
modeld_logs = trim_logs_to_max_frames(lr, MAX_FRAMES, {"roadCameraState", "wideRoadCameraState"}, {"roadEncodeIdx", "wideRoadEncodeIdx"})
dmodeld_logs = trim_logs_to_max_frames(lr, MAX_FRAMES, {"driverCameraState"}, {"driverEncodeIdx"})
@ -128,18 +119,9 @@ def model_replay(lr, frs):
modeld = get_process_config("modeld")
dmonitoringmodeld = get_process_config("dmonitoringmodeld")
try:
if spinner:
spinner.update("running model replay")
modeld_msgs = replay_process(modeld, modeld_logs, frs)
dmonitoringmodeld_msgs = replay_process(dmonitoringmodeld, dmodeld_logs, frs)
log_msgs.extend([m for m in modeld_msgs if m.which() == "modelV2"])
log_msgs.extend([m for m in dmonitoringmodeld_msgs if m.which() == "driverStateV2"])
finally:
if spinner:
spinner.close()
return log_msgs
modeld_msgs = replay_process(modeld, modeld_logs, frs)
dmonitoringmodeld_msgs = replay_process(dmonitoringmodeld, dmodeld_logs, frs)
return modeld_msgs + dmonitoringmodeld_msgs
if __name__ == "__main__":
@ -196,8 +178,8 @@ if __name__ == "__main__":
cmp_log = []
# logs are ordered based on type: modelV2, driverStateV2, nav messages (navThumbnail, mapRenderState, navModel)
model_start_index = next(i for i, m in enumerate(all_logs) if m.which() == "modelV2")
cmp_log += all_logs[model_start_index:model_start_index + MAX_FRAMES]
model_start_index = next(i for i, m in enumerate(all_logs) if m.which() in ("modelV2", "cameraOdometry"))
cmp_log += all_logs[model_start_index:model_start_index + MAX_FRAMES*2]
dmon_start_index = next(i for i, m in enumerate(all_logs) if m.which() == "driverStateV2")
cmp_log += all_logs[dmon_start_index:dmon_start_index + MAX_FRAMES]
if not NO_NAV:

@ -1 +1 @@
2af877ca0ce6996a4c1cf54cd11abf5a1f4e0576
3bb23e270a3a7219cfeedadd8d085c32fc571c0d

Loading…
Cancel
Save