ComposeDataset

class paddle.io. ComposeDataset [源代码]

由多个数据集的字段组成的数据集。

这个数据集用于将多个映射式(map-style)且长度相等数据集按字段组合为一个新的数据集。

参数

  • datasets (list of Dataset) - 待组合的多个数据集。

返回

Dataset,字段组合后的数据集。

代码示例

>>> import numpy as np
>>> import paddle
>>> from paddle.io import Dataset, ComposeDataset


>>> # define a random dataset
>>> class RandomDataset(Dataset):
...     def __init__(self, num_samples):
...         self.num_samples = num_samples
...
...     def __getitem__(self, idx):
...         image = np.random.random([32]).astype('float32')
...         label = np.random.randint(0, 9, (1, )).astype('int64')
...         return image, label
...
...     def __len__(self):
...         return self.num_samples
...
>>> dataset = ComposeDataset([RandomDataset(10), RandomDataset(10)])
>>> for i in range(len(dataset)):
...     image1, label1, image2, label2 = dataset[i]
...     # do something