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])