图片使用resize_image_with_crop_or_pad方法报错问题

在学习tensorflow的过程中遇到一个图片预处理的问题:

具体代码如下:

image_raw_data = tf.gfile.FastGFile("./dog.jpeg",'rb').read()


with tf.Session() as sess:
    img_data = tf.image.decode_jpeg(image_raw_data)
    print (img_data.get_shape())
    # 输出解码之后的三维矩阵。
    #print(img_data.eval())
    img_data.set_shape([560, 735, 3])
    print (img_data)
    ?img_data.set_shape
    print(img_data.get_shape())
    
with tf.Session() as sess:
    croped = tf.image.resize_image_with_crop_or_pad(img_data,1000,1000)
    padded = tf.image.resize_image_with_crop_or_pad(img_data,3000,3000)
    plt.imshow(croped.eval())
    plt.show()
    plt.imshow(padded.eval())
    plt.show()

该段程序在ipython中运行报如下错误:

InvalidArgumentError: Expected size[1] in [0, 551], but got 600
	 [[Node: resize_image_with_crop_or_pad_44/crop_to_bounding_box/Slice = Slice[Index=DT_INT32, T=DT_UINT8, _device="/job:localhost/replica:0/task:0/device:CPU:0"](resize_image_with_crop_or_pad_44/ExpandDims, resize_image_with_crop_or_pad_44/crop_to_bounding_box/stack, resize_image_with_crop_or_pad_44/crop_to_bounding_box/stack_1)]]

上述报错英文暂时不太能看明白,但是经过自己的测试找到了原因,

上述的图片实际像素为551X735的大小。在这里我们是设置成了560X735,这个时候在对图片数据调用resize_image_with_crop_or_pad(img_data,1000,1000)则会报错,将上述img_data.set_shape([560, 735, 3])改成img_data.set_shape([551, 735, 3])(即大小在图片实际大小范围之内),则报错解除。




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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值