Tensor must be 4-D with last dim 1, 3, or 4,bug记录

最新推荐文章于 2024-07-29 19:21:03 发布

随手笔记_000333

最新推荐文章于 2024-07-29 19:21:03 发布

阅读量1.9k

点赞数

分类专栏： tensorflow 文章标签： bug 图片深度学习 tensorflow pytorch

本文链接：https://blog.csdn.net/superyr_210/article/details/79964498

版权

在学习PyTorch的过程中遇到坑，使用torchvision.utils.make_grid处理图片后得到的Tensor缺少batch size维度，不适用于TensorFlow的tf.summary.image。解决方案是利用tf.expand_dims增加维度。此外，文章还提到了torchvision.utils.save_image的用法。

摘要由CSDN通过智能技术生成

torch学习入坑（1）
记录一下第一个坑，防止自己忘记。
在用torchvision.utils.make_grid处理完图片之后，维度会变为三维，如果需要使用tf.summary.image ，需要增加batch size 的那一维，需要用到的函数为 tf.expand_dims。

注意：

1、torchvision.utils.make_grid函数
输入：Tensor of shape (B x C x H x W)
输出：多个图片拼接成的一个大图三维没有batch size 那一维
like this:
batch size 16

以下解释来自官网
链接http://pytorch.org/docs/master/torchvision/utils.html
torchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)
scale_each=False, pad_value=0)

作用：Make a grid of images.
Parameters:

    tensor (Tensor or list) – 4D mini-batch Tensor of shape (B x C x H x W) or a list of images all of the same size.
    nrow (int, optional) – Number of images displayed in each row of the grid. The Final grid size is (B / nrow, nrow). Default is 8.
    padding (int, optional) – amount of padding. Default is 2.
    normalize (bool, optional) – If True, shift the image to the range (0, 1), by subtracting the minimum and dividing by the maximum pixel value.
    range (tuple, optional) – tuple (min, max) where min and max are numbers, then these numbers are used to normalize the image. By default,