diff --git a/inference_utils.py b/inference_utils.py index 1e92fd4..d651dc0 100644 --- a/inference_utils.py +++ b/inference_utils.py @@ -31,7 +31,7 @@ class VideoReader(Dataset): 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 @@ class ImageSequenceWriter: def close(self): pass - \ No newline at end of file +