VOC2012

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

VOC2012 数据集

参数

  • 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 。

返回

VOC2012数据集实例

代码示例

  1. import paddle
  2. from paddle.vision.datasets import VOC2012
  3. from paddle.vision.transforms import Normalize
  5. class SimpleNet(paddle.nn.Layer):
  6.     def __init__(self):
  7.         super(SimpleNet, self).__init__()
  9.     def forward(self, image, label):
  10.         return paddle.sum(image), label
  12. normalize = Normalize(mean=[0.5, 0.5, 0.5],
  13.                       std=[0.5, 0.5, 0.5],
  14.                       data_format='HWC')
  15. voc2012 = VOC2012(mode='train', transform=normalize, backend='cv2')
  17. for i in range(10):
  18.     image, label= voc2012[i]
  19.     image = paddle.cast(paddle.to_tensor(image), 'float32')
  20.     label = paddle.to_tensor(label)
  22.     model = SimpleNet()
  23.     image, label= model(image, label)
  24.     print(image.numpy().shape, label.numpy().shape)