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