|
@@ -31,7 +31,7 @@ class VideoReader(Dataset):
|
|
class VideoWriter:
|
|
class VideoWriter:
|
|
def __init__(self, path, frame_rate, bit_rate=1000000):
|
|
def __init__(self, path, frame_rate, bit_rate=1000000):
|
|
self.container = av.open(path, mode='w')
|
|
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.pix_fmt = 'yuv420p'
|
|
self.stream.bit_rate = bit_rate
|
|
self.stream.bit_rate = bit_rate
|
|
|
|
|
|
@@ -85,4 +85,4 @@ class ImageSequenceWriter:
|
|
|
|
|
|
def close(self):
|
|
def close(self):
|
|
pass
|
|
pass
|
|
-
|
|
|
|
|
|
+
|