Flowers

class paddle.vision.datasets. Flowers [源代码]

Flowers102 数据集

参数

  • data_file (str) - 数据文件路径,如果 download 参数设置为 True , data_file 参数可以设置为 None 。默认值为 None 。

  • label_file (str) - 标签文件路径,如果 download 参数设置为 True , label_file 参数可以设置为 None 。默认值为 None 。

  • setid_file (str) - 子数据集下标划分文件路径,如果 download 参数设置为 True , setid_file 参数可以设置为 None 。默认值为 None 。

  • mode (str) - 'train' 或 'test' 模式,默认为 'train' 。

  • transform (callable) - 图片数据的预处理,若为 None 即为不做预处理。默认值为 None。

  • download (bool) - 当 data_file 是 None 时,该参数决定是否自动下载数据集文件。默认为 True 。

  • backend (str, optional) - 指定要返回的图像类型:PIL.Image或numpy.ndarray。必须是{'pil','cv2'}中的值。如果未设置此选项,将从paddle.vsion.get_image_backend获得这个值。 默认值: None 。

返回

Flowers数据集实例

代码示例

from paddle.vision.datasets import Flowers

flowers = Flowers(mode='test')

for i in range(len(flowers)):
    sample = flowers[i]
    print(sample[0].size, sample[1])