PaddlePaddle如何输入数据不指定尺寸

  • 问题描述:PaddlePaddle的Fluid版输入数据是否可以不指定尺寸,或者实际输入图像跟所指定尺寸不一样?

  • 报错输出:

Traceback (most recent call last):
  File "/Users/jizhi/Desktop/Paddle/Paddlecode/test.py", line 4, in <module>
    images = fluid.layers.data(name='pixel', shape=None, dtype='float32')
  File "/Users/jizhi/anaconda3/envs/paddle/lib/python3.5/site-packages/paddle/fluid/layers/io.py", line 77, in data
    shape = list(shape)
TypeError: 'NoneType' object is not iterable
  • 问题复现:
import paddle.fluid as fluid
import numpy as np

images = fluid.layers.data(name='pixel', shape=None, dtype='float32')
  • 问题分析:

在PaddlePaddle中,data layer是必须指定shape尺寸的,但实际输入叙事指定尺寸可以不一样,如果data layer没有定义shape,则会报出'NoneType' object is not iterable

  • 问题解答:

使用fluid.lyaers.data()方法时需要指定shape参数

images = fluid.layers.data(name='images', shape=[3, 32, 32], dtype='float32')
  • 问题拓展:

data layer中shape是必须添加的,可以从PaddlePaddle的data()方法代码看出这一样要求。

def data(name,
         shape,
         append_batch_size=True,
         dtype='float32',
         lod_level=0,
         type=core.VarDesc.VarType.LOD_TENSOR,
         stop_gradient=True):
    helper = LayerHelper('data', **locals())
    shape = list(shape)
    for i in six.moves.range(len(shape)):
        if shape[i] is None:
            shape[i] = -1
            append_batch_size = False
        elif shape[i] < 0:
            append_batch_size = False

    if append_batch_size:
        shape = [-1] + shape  # append batch size as -1

    data_var = helper.create_global_variable(
        name=name,
        shape=shape,
        dtype=dtype,
        type=type,
        stop_gradient=stop_gradient,
        lod_level=lod_level,
        is_data=True)
    return data_var

从data()方法的代码中可以看出,如果shape中没有定义batch,会自动设置为-1。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值