首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >Tensorflow Assign需要两个张量的形状才能匹配。lhs shape= [20] rhs shape= [48]

Tensorflow Assign需要两个张量的形状才能匹配。lhs shape= [20] rhs shape= [48]
EN

Stack Overflow用户
提问于 2016-11-15 12:19:34
回答 6查看 26.5K关注 0票数 15

我是TensorFlow新手。我已经从deeppose的开源实现中训练了一个TensorFlow模型,现在必须对一组新的图像运行该模型。

该模型是在100 * 100大小的图像上训练的,因此我将新图像集的大小调整为相同的大小。我有要运行模型的149新图像。当我运行模型时,我得到以下错误。

代码语言:javascript
复制
InvalidArgumentError (see above for traceback): Assign requires shapes
of both tensors to match. lhs shape= [20] rhs shape= [48]

在一线

代码语言:javascript
复制
saver = tf.train.Saver(tf.all_variables())

我怀疑训练过的模型大小和测试图像大小不匹配。我不清楚如何解决这个问题。我打印出了tf.all_variables()调用的变量列表。这就是了

代码语言:javascript
复制
Tensor("Placeholder:0", shape=(128, 100, 100, 3), dtype=float32)
(11, 11, 3, 20)
conv1/weights:0
(20,)
conv1/biases:0
(5, 5, 20, 35)
conv2/weights:0
(35,)
conv2/biases:0
(3, 3, 35, 50)
conv4/weights:0
(50,)
conv4/biases:0
(3, 3, 50, 75)
conv5/weights:0
(75,)
conv5/biases:0
(300, 1024)
local1/weights:0
(1024,)
local1/biases:0
(1024, 1024)
local2/weights:0
(1024,)
local2/biases:0
(1024, 0)
softmax_linear/weights:0
(0,)
softmax_linear/biases:0

我不确定RHS参数是从哪里来的。我查看了所有的配置文件,似乎没有任何参数指定此配置。

任何帮助解决这个问题的人都将不胜感激。

EN

Stack Overflow用户

发布于 2018-04-12 00:16:09

我也遇到了这个问题,问题是标签和类编号不匹配,所以我更改并修复了类编号计数和到处都是的标签。

在我的例子中,它更改了faster_rcnn.config和"label_map.pbtxt“文件中的"num_classes”参数,使其与实际值相匹配。

票数 3
EN
查看全部 6 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/40601975

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档