Change video encoding

mikgroup · Sep 10, 2019 · cecb033 · cecb033
1 parent 8e290b8
commit cecb033
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/sigpy/plot.py b/sigpy/plot.py
@@ -325,9 +325,16 @@ def key_press(self, event):
                                  format='png', bbox_inches=bbox, pad_inches=0)
 
             subprocess.run(['ffmpeg',
-                            '-framerate', str(self.fps),
+                            '-f', 'image2',
+                            '-s', '{}x{}'.format(
+                                int(bbox.width * self.fig.dpi),
+                                int(bbox.height * self.fig.dpi)),
+                            '-r', str(self.fps),
                             '-i', '{} %05d.png'.format(temp_basename),
-                            '-vcodec', 'png',
+                            '-vf', 'scale=trunc(iw/2)*2:trunc(ih/2)*2',
+                            '-vcodec', 'libx264',
+                            '-preset', 'veryslow',
+                            '-pix_fmt', 'yuv420p',
                             filename])
 
             for i in range(self.shape[self.d]):