lfolle commited on
Commit
3fe56b6
1 Parent(s): 3ae87b7

Change to h264 mp4.

Browse files
Files changed (1) hide show
  1. create_video.py +5 -3
create_video.py CHANGED
@@ -16,7 +16,8 @@ def noise_process(numpy_image, steps=149):
16
 
17
 
18
  def generate_video(numpy_image):
19
- save_path = "result.ogg"
 
20
  fps = 30
21
  sec = 5
22
  image_lst = noise_process(numpy_image)
@@ -30,6 +31,7 @@ def generate_video(numpy_image):
30
  for i, img in enumerate(image_lst):
31
  Image.fromarray(img).save(f"video/{i:03d}.jpg", quality=95)
32
 
33
- ffmpeg.input('video/*.jpg', pattern_type='glob').output(save_path, qscale=10).run(overwrite_output=True)
34
- # ffmpeg.input('video/%3d.jpg').output(save_path, qscale=10).run(overwrite_output=True)
 
35
  return save_path
16
 
17
 
18
  def generate_video(numpy_image):
19
+ # save_path = "result.ogg"
20
+ save_path = "result.mp4"
21
  fps = 30
22
  sec = 5
23
  image_lst = noise_process(numpy_image)
31
  for i, img in enumerate(image_lst):
32
  Image.fromarray(img).save(f"video/{i:03d}.jpg", quality=95)
33
 
34
+ # ffmpeg.input('video/*.jpg', pattern_type='glob').output(save_path, qscale=10).run(overwrite_output=True)
35
+ # ffmpeg.input('video/%3d.jpg').output(save_path, crf=5, vcodec="h264").run(overwrite_output=True)
36
+ ffmpeg.input('video/*.jpg', pattern_type='glob').output(save_path, crf=5, vcodec="h264").run(overwrite_output=True)
37
  return save_path