首页 Paddle框架 帖子详情
请问下这个data层能否输入长宽信息?
收藏
快速回复
Paddle框架 问答深度学习模型训练 981 2
请问下这个data层能否输入长宽信息?
收藏
快速回复
Paddle框架 问答深度学习模型训练 981 2

请问下这个data层能否输入长宽信息?

image = paddle.layer.data(name="image", type=paddle.data_type.dense_vector(DATA_DIM))      发现如果输入是长方形的时候paddle.layer.img_conv函数会报错

paddle.layer.img_conv这个函数如果没有输入长宽信息,就会把一个图片向量看成一个正方形,宽高为sqrt(dim),难道不支持对长方形图片做卷积?

0
收藏
回复
全部评论(2)
时间顺序
挪威的仙人掌
#2 回复于2018-05

可以输入长宽信息,参考:

https://github.com/PaddlePaddle/Paddle/blob/develop/python/paddle/trainer_config_helpers/layers.py#L933

0
回复
夜雨飘零1
#3 回复于2018-05

这有个例子:https://github.com/PaddlePaddle/models/blob/develop/scene_text_recognition/network_conf.py#L36-L37

        self.image = layer.data(
            name='image',
            type=paddle.data_type.dense_vector(self.image_vector_size),
            height=self.shape[0],
            width=self.shape[1])
0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户