提交 0186db31 编写于 作者: D dengkaipeng

add return description. test=develop

上级 36d2fd4d
...@@ -18,7 +18,7 @@ Cifar100 ...@@ -18,7 +18,7 @@ Cifar100
返回 返回
::::::::: :::::::::
Cifar100数据集实例 Cifar100数据集实例,包含imagelabel两个字段,image数据形状为[3, 32, 32]label数据形状为[1]
代码示例 代码示例
::::::::: :::::::::
......
...@@ -18,7 +18,7 @@ Cifar10 ...@@ -18,7 +18,7 @@ Cifar10
返回 返回
::::::::: :::::::::
Cifar10数据集实例 Cifar10数据集实例,包含imagelabel两个字段,image数据形状为[3, 32, 32]label数据形状为[1]
代码示例 代码示例
::::::::: :::::::::
......
...@@ -20,7 +20,7 @@ Flowers ...@@ -20,7 +20,7 @@ Flowers
返回 返回
::::::::: :::::::::
Flowers数据集实例 Flowers数据集实例,包含image和label两个字段,image数据形状为图像尺寸,label数据形状为[1]
代码示例 代码示例
::::::::: :::::::::
......
...@@ -18,7 +18,7 @@ MNIST ...@@ -18,7 +18,7 @@ MNIST
返回 返回
::::::::: :::::::::
MNIST数据集实例 MNIST数据集实例,包含image和label两个字段,image数据形状为[1, 28, 28], label数据形状为[1]。
代码示例 代码示例
::::::::: :::::::::
......
...@@ -20,7 +20,7 @@ VOC2012 ...@@ -20,7 +20,7 @@ VOC2012
返回 返回
::::::::: :::::::::
VOC2012数据集实例 VOC2012数据集实例,包含imagelabel两个字段,数据形状均为图像尺寸。
代码示例 代码示例
::::::::: :::::::::
......
.. _cn_api_paddle_vision_datasets_DatasetFolder:
DatasetFolder
-------------------------------
.. py:class:: paddle.vision.datasets.DatasetFolder(root, loader=None, extensions=None, transform=None, is_valid_file=None)
一种通用的数据加载方式,当输入以如下的格式存放时:
root/class_a/1.ext
root/class_a/2.ext
root/class_a/3.ext
root/class_b/123.ext
root/class_b/456.ext
root/class_b/789.ext
参数:
- **root** (str) - 根目录路径。
- **loader** (callable,可选) - 可以加载数据路径的一个函数,如果该值没有设定,默认使用 ``cv2.imread`` 。默认值:None。
- **extensions** (tuple[str],可选) - 允许的数据后缀列表,如果该值没有设定,默认使用 ``('.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm', '.tif', '.tiff', '.webp')`` 。默认值:None。
- **transform** (callable,可选) - 数据增强函数。默认值:None。
- **is_valid_file** (callable,可选) - 根据每条数据的路径来判断是否合法的一个函数。默认值:None。
**代码示例**:
.. code-block:: python
import os
import cv2
import tempfile
import shutil
import numpy as np
from paddle.vision.datasets import DatasetFolder
def make_fake_dir():
data_dir = tempfile.mkdtemp()
for i in range(2):
sub_dir = os.path.join(data_dir, 'class_' + str(i))
if not os.path.exists(sub_dir):
os.makedirs(sub_dir)
for j in range(2):
fake_img = (np.random.random((32, 32, 3)) * 255).astype('uint8')
cv2.imwrite(os.path.join(sub_dir, str(j) + '.jpg'), fake_img)
return data_dir
temp_dir = make_fake_dir()
data_folder = DatasetFolder(temp_dir)
for items in data_folder:
break
shutil.rmtree(temp_dir)
.. _cn_api_paddle_vision_datasets_ImageFolder:
ImageFolder
-------------------------------
.. py:class:: paddle.vision.datasets.ImageFolder(root, loader=None, extensions=None, transform=None, is_valid_file=None)
一种通用的数据加载方式,当输入以如下的格式存放时:
root/1.ext
root/2.ext
root/sub_dir/3.ext
参数:
- **root** (str) - 根目录路径。
- **loader** (callable,可选) - 可以加载数据路径的一个函数,如果该值没有设定,默认使用 ``cv2.imread`` 。默认值:None。
- **extensions** (tuple[str],可选) - 允许的数据后缀列表,如果该值没有设定,默认使用 ``('.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm', '.tif', '.tiff', '.webp')`` 。默认值:None。
- **transform** (callable,可选) - 数据增强函数。默认值:None。
- **is_valid_file** (callable,可选) - 根据每条数据的路径来判断是否合法的一个函数。默认值:None。
**代码示例**:
.. code-block:: python
import os
import cv2
import tempfile
import shutil
import numpy as np
from paddle.vision.datasets import ImageFolder
def make_fake_dir():
data_dir = tempfile.mkdtemp()
for i in range(2):
sub_dir = os.path.join(data_dir, 'class_' + str(i))
if not os.path.exists(sub_dir):
os.makedirs(sub_dir)
for j in range(2):
fake_img = (np.random.random((32, 32, 3)) * 255).astype('uint8')
cv2.imwrite(os.path.join(sub_dir, str(j) + '.jpg'), fake_img)
return data_dir
temp_dir = make_fake_dir()
data_folder = ImageFolder(temp_dir)
for items in data_folder:
break
shutil.rmtree(temp_dir)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册