Update inference_utils.py

Improves accuracy of frame rate. See PyAV-Org/PyAV#580
Liam203 · Mar 10, 2023 · 3d94959 · 3d94959
1 parent 81a1093
commit 3d94959
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/inference_utils.py b/inference_utils.py
@@ -31,7 +31,7 @@ def __getitem__(self, idx):
 class VideoWriter:
     def __init__(self, path, frame_rate, bit_rate=1000000):
         self.container = av.open(path, mode='w')
-        self.stream = self.container.add_stream('h264', rate=round(frame_rate))
+        self.stream = self.container.add_stream('h264', rate=f'{frame_rate:.4f}')
         self.stream.pix_fmt = 'yuv420p'
         self.stream.bit_rate = bit_rate
 
@@ -85,4 +85,4 @@ def write(self, frames):
 
     def close(self):
         pass
-
+