Python随机打乱训练图像数据和标签

2021-01-09  本文已影响0人  LabVIEW_Python

假设有训练数据 train_images和标签train_labels,已经转换为np.ndarray类型,有两种方式实现训练图像数据和标签的随机打乱。

# 在每轮迭代开始前,将训练数据随机打乱
seed_id = np.random.randint(1,100)
np.random.seed(seed_id)
np.random.shuffle(train_images)
np.random.seed(seed_id)
np.random.shuffle(train_labels)
# 在每轮迭代开始前,将训练数据随机打乱
index = [i for i in range(train_images.shape[0])]
np.random.shuffle(index) # 打乱索引
image = train_images[index]
label = train_label[index]
上一篇 下一篇

猜你喜欢

热点阅读