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

Posted 2021-07-21 Tina姐

在使用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, 我们下篇博客见~

最后，求点赞，评论，收藏。或者一键三连