0
点赞
收藏
分享

微信扫一扫

将Moving-MNIST数据集处理成图片文件

RockYoungTalk 2022-04-05 阅读 59

Moving-MNIST数据集下载地址: 

http://www.cs.toronto.edu/~nitish/unsupervised_video/icon-default.png?t=M276http://www.cs.toronto.edu/~nitish/unsupervised_video/ 数据集处理:

将下载的数据集处理成图片文件!

# 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('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='C:/Users/liq/Desktop/AIcode/LMC-Memory-main/movingmnist/train/video_%d/frame_%d.jpg'%(index,count)
        count+=1
        savepath='C:/Users/liq/Desktop/AIcode/LMC-Memory-main/movingmnist/train/video_%d'%(index)
        if not os.path.exists(savepath):
            os.mkdir(savepath)
        cv2.imwrite(filename,img)
        # cv2.imshow('img0', img)
        # cv2.waitKey(0)


处理后格式:
movingmnist
├── train
│   ├── video_00000
│   │   ├── frame_00000.jpg
...
│   │   ├── frame_xxxxx.jpg
...
│   ├── video_xxxxx

举报

相关推荐

0 条评论