Script to extract thumbnails from segment (#22294)

* Script to extract thumnails from segment

* fix name

* whitespace
old-commit-hash: 8a836b7e1e
commatwo_master
Willem Melching 4 years ago committed by GitHub
parent 4827dea04c
commit cfbee2bdca
  1. 1
      selfdrive/camerad/test/.gitignore
  2. 28
      selfdrive/camerad/test/get_thumbnails_for_segment.py

@ -0,0 +1,28 @@
#!/usr/bin/env python3
import os
from tqdm import tqdm
from common.file_helpers import mkdirs_exists_ok
from tools.lib.logreader import LogReader
from tools.lib.route import Route
import argparse
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("route", help="The route name")
parser.add_argument("segment", type=int, help="The index of the segment")
args = parser.parse_args()
out_path = os.path.join("jpegs", f"{args.route.replace('|', '_')}_{args.segment}")
mkdirs_exists_ok(out_path)
r = Route(args.route)
lr = list(LogReader(r.qlog_paths()[args.segment]))
for msg in tqdm(lr):
if msg.which() == 'thumbnail':
with open(os.path.join(out_path, f"{msg.thumbnail.frameId}.jpg"), 'wb') as f:
f.write(msg.thumbnail.thumbnail)
Loading…
Cancel
Save