诗人的张量流:“名称'导入/输入'是指图中没有的操作。”
machine-learning
python
tensorflow
7
0

我遵循了针对诗人的代码实验室张量流,并且训练工作正常,但是当我运行脚本来评估图像时:

python -m scripts.label_image \
    --graph=tf_files/retrained_graph.pb  \
    --image=tf_files/flower_photos/daisy/21652746_cc379e0eea_m.jpg

我收到以下错误:

The name 'import/input' refers to an Operation not in the graph.

我环顾四周,这与选择输入和输出层有关,脚本label_image.py已将“ input”和“ output”设置为默认值。我正在使用的架构是“ inception_v3”。

参考资料:
Stack Overflow
收藏
评论
共 5 个回答
高赞 时间 活跃

正如@Mimii和@Celio提到的那样:将〜/ scripts / label_image.py更改为input_layer = "input"input_layer = "Mul" ,并更改输入尺寸: input_height = 299input_width= 299

收藏
评论

使用--input_layer名称作为占位符。因为retrain.py脚本已将input_layer的默认值设置为“ Placeholder”,所以它将起作用。

    python label_image.py 
          --graph=retrained_graph.pb 
          --labels=retrained_labels.txt 
          --output_layer=final_result 
          --image=testimage654165.jpg 
          --input_layer=Placeholder
收藏
评论

您必须在scripts文件夹中的label_image.py中进行一些更改

input_height = 299 input_height224更改为299
input_width = 299 input_width224更改为299
input_mean = 128
input_std = 128
input_layer = "Mul" input_layer输入更改为Mul
output_layer = "final_result"

输出:

Evaluation time (1-image): 1.901s

daisy (score=0.98584)
sunflowers (score=0.01136)
dandelion (score=0.00210)
tulips (score=0.00066)
roses (score=0.00004)

有关更多信息,请参阅此页面

收藏
评论

并非所有人都收到此错误。我猜测如果您使用MobileNet以外的任何其他体系结构,则会出现此错误。在您的label_image.py文件中,将值更改为:

input_height = 299
input_width = 299
input_layer = "Mul"

这应该解决。

收藏
评论

我更改了〜/ scripts / label_image.py第77行,它的工作原理是:

input_layer = "input"

input_layer = "Mul"
收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号