Python creating video from images using opencv

Question:

I was trying to create a video to show the dynamic variation of the data, like just continuously showing the images one by one quickly, so I used images (the images just called 1,2,3,4,…..) and wrote the following code:

import cv2
import numpy as np

img=[]
for i in range(0,5):
    img.append(cv2.imread(str(i)+'.png'))

height,width,layers=img[1].shape

video=cv2.VideoWriter('video.avi',-1,1,(width,height))

for j in range(0,5):
    video.write(img)

cv2.destroyAllWindows()
video.release()

and a error was raised:

TypeError: image is not a numpy array, neither a scalar

I think I used the list in a wrong way but I’m not sure. So where did I do wrong?

Asked By: Yingqiang Gao

||

Answers:

You are writing the whole array of frames. Try to save frame by frame instead:

...
for j in range(0,5):
  video.write(img[j])
...

reference

Answered By: Václav Struhár

You can read the frames and write them to video in a loop. Following is your code with a small modification to remove one for loop.

  import cv2
  import numpy as np
  
  # choose codec according to format needed
  fourcc = cv2.VideoWriter_fourcc(*'mp4v') 
  video = cv2.VideoWriter('video.avi', fourcc, 1, (width, height))

  for j in range(0,5):
     img = cv2.imread(str(i) + '.png')
     video.write(img)

  cv2.destroyAllWindows()
  video.release()

Alternatively, you can use skvideo library to create video form sequence of images.

  import numpy as np
  import skvideo.io
   
  out_video =  np.empty([5, height, width, 3], dtype = np.uint8)
  out_video =  out_video.astype(np.uint8)
  
  for i in range(5):
      img = cv2.imread(str(i) + '.png')
      out_video[i] = img

  # Writes the the output image sequences in a video file
  skvideo.io.vwrite("video.mp4", out_video)

  
Answered By: ravikt

You can use this pip package. It provides CLI commands to make video from images.

img_to_vid.py -f images_directory

Categories: questions Tags: ,
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.