bfshi commited on
Commit
357a135
·
1 Parent(s): 7e6904b

pixel-format-fix (#3)

Browse files

- update pixel format (0b97fc423c2a2b5ba36d8fec8c37b8e6fcff1a91)

Files changed (1) hide show
  1. demo_utils.py +2 -2
demo_utils.py CHANGED
@@ -39,7 +39,7 @@ def image_to_video(image_path, output_path, fps):
39
 
40
  img_array = np.array(img)
41
 
42
- with imageio.get_writer(output_path, fps=fps, format='FFMPEG', codec='libx264', pixelformat='yuv420p', macro_block_size=1) as writer:
43
  writer.append_data(img_array)
44
 
45
  return {
@@ -582,6 +582,6 @@ def process_video(video_path, setup, gazing_ratio=0.75, task_loss_requirement=0.
582
 
583
 
584
  def save_video(frames, output_path, fps):
585
- with imageio.get_writer(output_path, fps=fps, format='FFMPEG', codec='libx264', pixelformat='yuv420p', macro_block_size=1) as writer:
586
  for frame in frames:
587
  writer.append_data(frame)
 
39
 
40
  img_array = np.array(img)
41
 
42
+ with imageio.get_writer(output_path, fps=fps, format='FFMPEG', codec='libx264', pixelformat='yuv444p', macro_block_size=1) as writer:
43
  writer.append_data(img_array)
44
 
45
  return {
 
582
 
583
 
584
  def save_video(frames, output_path, fps):
585
+ with imageio.get_writer(output_path, fps=fps, format='FFMPEG', codec='libx264', pixelformat='yuv444p', macro_block_size=1) as writer:
586
  for frame in frames:
587
  writer.append_data(frame)