소스 검색

Merge pull request #227 from dcyoung/master

Improves accuracy of frame rate
Peter Lin 2 년 전
부모
커밋
53d74c6826
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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
-        
+