mawei/sunnypilot
1
0
Fork 0
mirror of https://github.com/sunnypilot/sunnypilot.git synced 2026-02-26 10:23:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

tools/rerun: Rerun 0.18 (#33188)

Browse Source
This commit is contained in:
Hoang Bui
2024-08-08 17:30:14 -04:00
committed by GitHub
parent ad33cae40e
commit a604dcc524
4 changed files with 75 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tools/rerun/camera_reader.py
View File

@@ -37,6 +37,7 @@ class _FrameReader:
frame_sz = self.w * self.h * 3 // 2
proc = subprocess.Popen(
["ffmpeg", "-v", "quiet", "-i", self.camera_path, "-f", "rawvideo", "-pix_fmt", "nv12", "-"],
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
stderr=subprocess.DEVNULL
)
@@ -84,9 +85,9 @@ class CameraReader:
def _run_on_segment(self, func, i):
return func(self._get_fr(i))
def run_across_segments(self, num_processes, func):
def run_across_segments(self, num_processes, func, desc=None):
with multiprocessing.Pool(num_processes) as pool:
num_segs = len(self.seg_idxs)
for _ in tqdm.tqdm(pool.imap_unordered(partial(self._run_on_segment, func), self.seg_idxs), total=num_segs):
for _ in tqdm.tqdm(pool.imap_unordered(partial(self._run_on_segment, func), self.seg_idxs), total=num_segs, desc=desc):
continue