Kaynağa Gözat

Merge pull request #227 from dcyoung/master

Improves accuracy of frame rate
Peter Lin 2 yıl önce
ebeveyn
işleme
53d74c6826
1 değiştirilmiş dosya ile 2 ekleme ve 2 silme
  1. 2 2
      inference_utils.py

+ 2 - 2
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
-        
+