Script to extract thumbnails from segment (#22294)
* Script to extract thumnails from segment
* fix name
* whitespace
old-commit-hash: 8a836b7e1e
commatwo_master
parent
4827dea04c
commit
cfbee2bdca
2 changed files with 29 additions and 0 deletions
@ -0,0 +1 @@ |
||||
jpegs/ |
@ -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…
Reference in new issue