Видеозапись, анимация - PullRequest
0 голосов
/ 24 февраля 2020

Я пытаюсь написать / сделать видео / анимацию в python. я сгенерировал ".png" изображения, используя double для l oop, я хочу сохранить изображения в массиве, а затем использовать их в видеозаписи без сохранения изображений на локальный диск.

Вот мой код

import math
import cv2
import matplotlib.pyplot as plt
from PIL import Image, ImageDraw
l1 = 1
l2 = 0.5
n_theta = 10
theta_start = 0
theta_end = math.pi/2
theta1 = []
theta2 = []
img_array = []
fig = plt.figure()

for i in range(0,n_theta):
    tmp = theta_start + i*(theta_end - theta_start)/(n_theta-1)
    theta1.append(tmp)
    theta2.append(tmp)

x0 = 0
y0 = 0
ct = 1

for t1 in theta1:
    for t2 in theta2:

        x1 = l1*math.cos(t1)
        y1 = l1*math.sin(t1)

        x2 = x1 + l2*math.cos(t2)
        y2 = y1 + l2*math.sin(t2)

        plt.figure()
        plt.plot([x0,x1],[y0,y1],[x1,x2],[y1,y2])
        plt.axis([0,1.6,0,1.6])

        img_array.append(cv2.imread(str(ct) +'.png'))
        ct = ct + 1


fourcc = cv2.VideoWriter_fourcc(*'mp4v') 
video=cv2.VideoWriter('video.mp4', fourcc, 30,(432,288))

for j in range(0,len(img_array)):
    video.write(img_array[j])

cv2.destroyAllWindows()
video.release()

Пожалуйста, помогите мне написать видео без сохранения изображений 'png' на локальный диск

...