mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-04-08 02:23:55 +08:00
100f89a161
* openpilot v0.9.9 release date: 2025-06-05T19:54:08 master commit: 8aadf02b2fd91f4e1285e18c2c7feb32d93b66f5 * AGNOS 12.4 (#35558) agnos12.4 --------- Co-authored-by: Vehicle Researcher <user@comma.ai> Co-authored-by: Maxime Desroches <desroches.maxime@gmail.com>
32 lines
957 B
Python
Executable File
32 lines
957 B
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import glob
|
|
import onnx
|
|
|
|
BASEDIR = os.path.abspath(os.path.join(os.path.dirname(os.path.realpath(__file__)), "../"))
|
|
MASTER_PATH = os.getenv("MASTER_PATH", BASEDIR)
|
|
MODEL_PATH = "/selfdrive/modeld/models/"
|
|
|
|
def get_checkpoint(f):
|
|
model = onnx.load(f)
|
|
metadata = {prop.key: prop.value for prop in model.metadata_props}
|
|
return metadata['model_checkpoint'].split('/')[0]
|
|
|
|
if __name__ == "__main__":
|
|
print("| | master | PR branch |")
|
|
print("|-| ----- | --------- |")
|
|
|
|
for f in glob.glob(BASEDIR + MODEL_PATH + "/*.onnx"):
|
|
# TODO: add checkpoint to DM
|
|
if "dmonitoring" in f:
|
|
continue
|
|
|
|
fn = os.path.basename(f)
|
|
master = get_checkpoint(MASTER_PATH + MODEL_PATH + fn)
|
|
pr = get_checkpoint(BASEDIR + MODEL_PATH + fn)
|
|
print(
|
|
"|", fn, "|",
|
|
f"[{master}](https://reporter.comma.life/experiment/{master})", "|",
|
|
f"[{pr}](https://reporter.comma.life/experiment/{pr})", "|"
|
|
)
|