将Movingmnist视频数据转化成图片数据

# -*- coding: utf-8 -*-
# @Time : 2022/11/19 19:44
# @File : video2image.py
# @Project : LMC-Memory-main

# contains 10,000 sequences each of length 20 showing 2 digits moving in a 64 x 64 frame.
# 10000 个视频，每个视频20帧，每帧宽64  高64
import numpy as np
import cv2
import os

data = np.load('./data/mnist_test_seq.npy')
data = data.transpose(1, 0, 2, 3)
print(data.shape)  # 10000,20,64,64

for index in range(len(data[:, 0, 0, 0])):
    # print(data[index,...].shape)
    count = 0
    for j in data[index, ...]:
        img = np.expand_dims(j, -1)
        print(img.shape)
        filename = './data/movingmnist/test/video_%s/frame_%s.jpg' % (
        str(index).zfill(5), str(count).zfill(5))
        count += 1
        savepath = './data/movingmnist/test/video_%s' % (str(index).zfill(5))
        if not os.path.exists(savepath):
            os.mkdir(savepath)
        cv2.imwrite(filename, img)
       

# Dataset preparation example:
movingmnist
├── train
│   ├── video_00000
│   │   ├── frame_00000.jpg
...
│   │   ├── frame_xxxxx.jpg
...
│   ├── video_xxxxx