VOC2012¶

class paddle.vision. VOC2012 [源代码] ¶: VOC2012 数据集

参数¶

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

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

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

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

transform (callable) - 作用于图片数据的transform，若未 None 即为无transform。

download (bool) - 是否自定下载数据集文件。默认为 True 。

返回¶

VOC2012数据集实例

代码示例¶

import paddle
from paddle.vision.datasets import VOC2012
from paddle.vision.transforms import Normalize

class SimpleNet(paddle.nn.Layer):
    def __init__(self):
        super(SimpleNet, self).__init__()

    def forward(self, image, label):
        return paddle.sum(image), label

normalize = Normalize(mean=[0.5, 0.5, 0.5],
                      std=[0.5, 0.5, 0.5],
                      data_format='HWC')
voc2012 = VOC2012(mode='train', transform=normalize, backend='cv2')

for i in range(10):
    image, label= voc2012[i]
    image = paddle.cast(paddle.to_tensor(image), 'float32')
    label = paddle.to_tensor(label)

    model = SimpleNet()
    image, label= model(image, label)
    print(image.numpy().shape, label.numpy().shape)

PaddlePaddle

paddle.vision / datasets / VOC2012

VOC2012¶

参数¶

返回¶

代码示例¶

PaddlePaddle

paddle

paddle.amp

paddle.callbacks

paddle.compat

paddle.distributed

paddle.distribution

paddle.fluid

paddle.io

paddle.jit

paddle.metric

paddle.nn

paddle.onnx

paddle.optimizer

paddle.regularizer

paddle.static

paddle.sysconfig

paddle.text

paddle.utils

paddle.vision

paddle.vision / datasets / VOC2012

VOC2012¶

参数¶

返回¶

代码示例¶