首页 > 其他分享 >成功解决[AssertionError: Input tensor input format are different]

成功解决[AssertionError: Input tensor input format are different]

时间:2023-06-14 22:38:24浏览次数:44  
标签:AssertionError different tensor format image add 64 input


在使用tensorboard writer.add_image时

writer.add_image('img/fixed_img', denorm(fixed_img.data), 0)

报如下错误

assert(len(tensor.shape) == len(input_format)), "size of input tensor and input format are different.

AssertionError: size of input tensor and input format are different. tensor shape: (128, 3, 64, 64), input_format: CHW

从报错信息来看, input tensor的维度是(128, 3, 64, 64),而 input_format的格式需要是 CHW。两者不匹配。

分析到这儿就明白了。input tensor虽然格式也是CHW, 但它还有一个batch维度,所以报错。

add_image只接收单一图像,你给它传一个batch数据自然是不行的

解决办法

add_images 替换add_image。就可以显示batch数据了

有人说,你这也太简单了嘛,还废话那么多。

但是,我更希望分享解决bug的过程,而不是结果。

文章持续更新,可以关注微信公众号【医学图像人工智能实战营】,一个关注于医学图像处理领域前沿科技的公众号。坚持已实践为主,手把手带你做项目,打比赛,写论文。凡原创文章皆提供理论讲解,实验代码,实验数据。只有实践才能成长的更快,关注我们,一起学习进步~

我是Tina, 我们下篇博客见~

标签:AssertionError,different,tensor,format,image,add,64,input
From: https://blog.51cto.com/u_16159492/6481696

相关文章

  • Tensorflow01-回归问题
    1线性回归就是给你一堆数据[[x0,y0],[x1,y1],[x2,y2]-----[xn,yn]]然后得出一个y=wx+b来,这里我们引入损失函数loss=\(\sum\)(wxi+b-yi)^2,然后我们就是最小化这个loss从而使得w'x+b'->y2梯度下降法在这里w'=w-lr*\({dy\overdw}\)在这里为什么是减呢,例如:x'=x-0.005*\({dy\ov......
  • TensorFlow实践——Softmax Regression
    SoftmaxRegression是Logistic回归在多分类上的推广,对于Logistic回归以及SoftmaxRegression的详细介绍可以参见:简单易学的机器学习算法——Logistic回归利用Theano理解深度学习——LogisticRegression深度学习算法原理——SoftmaxRegression下面的代码是利用TensorFlow基本API实......
  • 对DenseTensor进行Transpose
    ML.NET是微软推出的为.NET平台设计的深度学习库,通过这个东西(ModelBuilder)可以自己构建模型,并用于后来的推理与数据处理。虽然设计是很好的,但是由于现在的AI发展基本上都以python实现作为基础,未来这个东西的发展不好说,特别是模型构建部分。我个人认为,它提供的最有价值的场景......
  • tensorflow实现花分类
    1.花数据集数据集来自kaggle官网下载。分为五类花,每类花有1000张图片。下载方式可以参考我的https://www.cnblogs.com/wancy/p/17446715.html 2.图片大小分布图训练模型之前,我们会需要先分析数据集,由于此类数据集每类花的图片数量一样,是均衡的。训练模型之前,我们......
  • 基于Tensorflow的Faster-Rcnn的断点续训
    一、前言最近在学习目标检测,到github上找了一个开源的Faster-RCNN项目(Tensorflow),项目地址是:https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3根据网上的各种教程,模型训练还算顺利,不过这个项目缺少断点续训的功能。也就是中途误操作导致训练中止,就只能从头开始......
  • Python+tensorflow计算整数阶乘的方法与局限性
    本文代码主要演示tensorflow的基本用法。importtensorflowas#创建变量,保存计算结果start=tf.Variable(1,dtype=tf.int64)#初始化变量的opinit_op=tf.global_variables_initializer()#启用默认图withtf.Session()assess:#初始化变量sess.run(ini......
  • raise AssertionError(“Torch not compiled with CUDA enabled“)
    raiseAssertionError(“TorchnotcompiledwithCUDAenabled”)遇到的问题,就是GPUdriver以及cuda都装完了,pytorch装完之后想试试代码。结果报错,raiseAssertionError(“TorchnotcompiledwithCUDAenabled”)。解决方案就是打开pytorch官网,对照一下cuda版本和pytorch......
  • opencv与tensorrt安装
     opencv安装参考:https://blog.csdn.net/wxyczhyza/article/details/128968849 tensorrt安装:sudotar-xvfTensorRT-8.5.1.7.Linux.x86_64-gnu.cuda-11.8.cudnn8.6.tar.gz#添加环境变量vim~/.bashrcexportLD_LIBRARY_PATH=$LD_LIBRARY_PATH:/解压文件位置/TensorRT-8.......
  • Google colab 更改Tensorflow深度学习框架版本
    Googlecolab默认导入的tensorflow版本是2.0想,想更改tensorflow版本为1.x则需要加入%tensorflow_version1.ximporttensorflowastftf.__version__之后要进行colab的重启,即可完成版本改装,由于colab没有terminal,所以只能这么修改。欢迎登陆官网(附https://tensorflow.google.cn/)......
  • 深度学习项目之mnist手写数字识别实战(TensorFlow框架)
    mnist手写数字识别是所有深度学习开发者的必经之路,mnist数据集的图片十分简单,是二值化图像,像素个数为28x28。所以对于所有研究深度学习的开发者来说学会mnist数据集的模型十分有必要。以此为实例进行计算机视觉如何进行识别出图片中的数据。MNIST手写数字数据集来自美国国家标准与......