tensorflow里面声明图片的形状

TensorFlow图片尺寸设置技巧

最新推荐文章于 2024-10-11 17:22:08 发布

原创最新推荐文章于 2024-10-11 17:22:08 发布 · 378 阅读

0 ·

CC 4.0 BY-SA版权

tensorflow 专栏收录该内容

18 篇文章

订阅专栏

本文介绍在使用TensorFlow进行图片处理时，如何在不返回图片尺寸的操作后手动设置图片尺寸信息，确保编译时模型能正确理解输入形状。

tensorflow有时候要对图片进行裁剪拼接等操作，基本上内置函数在裁剪的时候都会隐形的返回图片形状的tensor，但是有时候默写函数不会返回图片的size，所以让我们在后面抓取图片的形状信息的时候会很麻烦，于是有必要在op不返回size的时候，声明size的信息。

# Set static shape so that tensorflow knows shape at compile time. 
img_crop.set_shape((crop_h, crop_w, 4))
label_crop.set_shape((crop_h,crop_w, 1))