8

How do I make an mp4 video from a set of png images with the module imageio? I've tried this:

import imageio
import glob
writer = imageio.get_writer('test.mp4', fps=20)
for png_path in glob.glob('*.png'):
    im = imageio.imread(png_path),
    writer.append_data(im[:, :, 1])
writer.close()

I've also tried replacing im[:, :, 1] with im. What am I doing wrong? I'm happy to use another module.

tommy.carstensen
  • 8,962
  • 15
  • 65
  • 108

2 Answers2

15

You don't need to modify the image through im[:, :, 1]. For example, the code below takes all images starting with name in a folder specified by path and creates a videofile called "test.mp4"

fileList = []
for file in os.listdir(path):
    if file.startswith(name):
        complete_path = path + file
        fileList.append(complete_path)

writer = imageio.get_writer('test.mp4', fps=20)

for im in fileList:
    writer.append_data(imageio.imread(im))
writer.close()

All images have to be the same size so you should resize them before appending them to the video file. You can modify the fps through fps, I just set it at 20 because I was following your code.

Mr K.
  • 1,064
  • 3
  • 19
  • 22
6

A better version of your code combined with Mr K. would be:

import imageio
import glob
import os
writer = imageio.get_writer('test.mp4', fps=20)
for file in glob.glob(os.path.join(path,f'{name}*.png')):
    im = imageio.imread(file)
    writer.append_data(im)
writer.close()

P.S. Don't forget to add image resizing code before appending them to the video file.

HKay
  • 355
  • 3
  • 7