запись numpy массивов как видео с openCV

Question

У меня есть CSV-файл, где каждая строка представляет собой 1d изображение со значениями 1 * 512 пикселей. Я загружаю эти изображения в массив numpy numOfImage * 512. Затем я хочу записать эти изображения в файл images.avi как видео, используя opencv 4.2. проблема в том, что я не могу создать видео из 1d изображения. если я установлю IMAGE_HEIGHT на 2, он генерирует его. Так что, если я это сделаю, то я не могу открыть созданный файл AVI.

images = getExternalData()

IMAGE_WIDHT  = images.shape[1]
IMAGE_HEIGHT = 1

output = cv2.VideoWriter('images.avi', cv2.CAP_FFMPEG, cv2.VideoWriter_fourcc('F','F','V','1'), 10, (IMAGE_WIDHT, IMAGE_HEIGHT), False)

print("converting...")

for image in images:
    output.write(image[np.newaxis, :])

output.release()
print("done")