added -pix_fmt yuv420p to ffmpeg args if codec is libx264

chunder · Mar 26, 2014 · d107e6b · d107e6b · Zulko · Mar 28, 2014
1 parent 011047f
commit d107e6b
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/moviepy/video/io/ffmpeg_writer.py b/moviepy/video/io/ffmpeg_writer.py
@@ -56,17 +56,22 @@ def __init__(self, filename, size, fps, codec="libx264",
                   bitrate=None, withmask=False):
 
         self.filename = filename
-        cmd = [ FFMPEG_BINARY, '-y',
+        cmd = (
+            [ FFMPEG_BINARY, '-y',
             "-f", 'rawvideo',
             "-vcodec","rawvideo",
             '-s', "%dx%d"%(size[0],size[1]),
             '-pix_fmt', "rgba" if withmask else "rgb24",
             '-r', "%.02f"%fps,
             '-i', '-', '-an',
-            '-vcodec', codec] + (
-            ['-b',bitrate] if (bitrate!=None) else []) + [
-            '-r', "%d"%fps,
-            filename ]
+            '-vcodec', codec]
+            + (['-b',bitrate] if (bitrate!=None) else [])
+
+            # http://trac.ffmpeg.org/ticket/658
+            + (['-pix_fmt', 'yuv420p'] if (codec == 'libx264') else [])
+
+            + [ '-r', "%d"%fps, filename ]
+            )
 
         self.proc = sp.Popen(cmd,stdin=sp.PIPE, stderr=sp.PIPE)
         self.proc.stderr.close()