数据随机化（打乱）

工具random.shuffle
不多说直接上代码

import numpy as np
import random

#导入数据
input_data=np.load('E:/VQA_data/input66.npy')
Y=np.load('E:/VQA_data/Y_.npy')

#数据洗牌
index_shuf = [i for i in range(len(input_data))]

random.shuffle(index_shuf)

# 一一取出 问题，答案，和图片
input_data66=np.zeros([215519, 4480])
input_data66 = [input_data[i] for i in index_shuf]
Y66=np.zeros([215519, 1000])
Y66 = [Y[i] for i in index_shuf]


#转回矩阵
input_data66=np.array(input_data66)
Y66=np.array(Y66)
#储存

np.save('E:/VQA_data/input_datawuxu.npy',input_data66)
np.save('E:/VQA_data/Ywuxu.npy',Y66)


#测试数据集制作
tx=input_data66[210000:215000]
ty=Y66[210000:215000]
np.save('E:/VQA_data/tx.npy',tx)
np.save('E:/VQA_data/ty.npy',ty)

OK····················································