问题标签 [tensorflow-slim]

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

0 投票
2 回答
1087 浏览

tensorflow - 正向运行时如何从 tensorflow slim 模型 VGG 中提取特征?

我已经使用 TensorFlow slim 模型 vgg 训练了一个分类模型,使用 CASIA(人脸识别数据集)作为训练数据集。我想通过使用 LFW 数据集来测试模型,这是一个人脸匹配任务。所以我需要提取像fc7/fc8这样的网络特征,而不是softmax层,并比较特征之间的距离,以确定它们是否是同一个人。如何提取苗条模型的特征?

这是训练代码的一部分。

0 投票
0 回答
423 浏览

python - 如何恢复 Tensorflow-Slim 预训练模型?

使用Tensorflow 1.3 slim 模型,我们可以访问许多预训练模型。通常他们进来三个。例如对于 MobileNet:

显然,访问权重的传统方法不起作用:

这将给出一个错误:

使用后端类Event_accumulator只会返回空数组:

有没有办法解决这个问题来查看权重?我见过有人尝试使用 Slim-git制作冻结模型.pbtensorboard

0 投票
1 回答
1937 浏览

tensorflow - 如何在张量流模型中找到可训练层的名称?

我正在尝试为我拥有的数据集微调tensorflow/slim resnet-v2-50模型中的最后几层。

我正在努力寻找可以训练的层的名称。在张量流模型中,有没有办法找到可训练的层的名称?有没有办法让这些名称有序,以便我可以选择最后几层进行训练?有没有办法从张量板上获取这些信息?

0 投票
0 回答
852 浏览

c# - C# TensorFlow inceptionModel(.ckpt) 恢复并制作冻结模型文件(.pb)

我想下载inception_resnet_v2并恢复它并为 C# tensorflow 程序制作一个冻结的模型文件。(.PB)。

参考:如何恢复 tensorflow inceptions 检查点文件(ckpt)?

Python代码在这里:

inception_resnet_v2

inception_resnet_v2_2016_08_30.ckpt [226MB]

inception_resnet_v2_2016_08_30.pb [6MB]

输出文件不正确。

如何获取完整的 .pb 文件?

制作冻结模型文件参考:github.com/migueldeicaza/TensorFlowSharp/issues/85#issuecomment-313590825

命令:

错误:AssertionError:输出/预测不在图中

0 投票
1 回答
753 浏览

tensorflow - 如何在 slim.learning.train 的 init_fn 中加载两个检查点

我想在使用时加载两个检查点slim.learning.train。例如,

问题是我只能在 model_path 中输入一个检查点文件。我想设置两个检查点。我认为可以有两种可能的解决方案:

  • 修改下面的assign_from_checkpoint_fn函数,tf.contrib.framework.assign_from_checkpoint_fn使 model_path 可以是检查点文件的列表
  • 合并之前的两个检查点。(我没有找到任何工具)

有人帮助我吗?

0 投票
0 回答
951 浏览

tensorflow - TF - 在 Tensorboard 上标准化和可视化热图

我有一个网络,它应该将一批 10 个大小为 100x100 的图像作为输入,并为每个图像生成 21 个热图。因此输出是一个张量 (10, 100, 100, 21)。

问题是我想可视化这个输出(例如用于张量板)并且每个 100x100 热图都需要标准化。如果输出的每张图像只有一个热图,事情就会很简单。

实现上述内容的有效方法是什么?

0 投票
1 回答
671 浏览

tensorflow - 在预测中使用 TensorFlow 对象检测 API 模型

我使用 TensorFlow对象检测 API从头开始​​训练 SSD Inception 模型。评估脚本表明该模型已经学到了一些东西,现在我想使用该模型。

我已经查看了可以将单个图像提供给经过训练的模型的对象检测ipynb 。但是,这适用于带有 MobileNet 的 SSD。我使用以下行(加载元图后)打印我训练的 TensorFlow 模型的张量名称。

但它不包含与 ipynb 中相同的输入或输出张量名称。我也搜索了代码,但是许多函数相互指向,很难找到我要找的东西。

如何找到我需要的张量名称?还是有另一种我不知道的方法?

0 投票
0 回答
46 浏览

tensorflow - tensorflow模型attention_ocr使用gtx1070训练时出现oom错误

我的 gpu 是 gtx1070,当我想训练 attention_ocr 模型时,日志出现错误 ResourceExhaustedError:OOM when allocating tensor with shape

文档说用gtx980测试过,所以我的gtx1070理论上是可以的。</p>

0 投票
1 回答
817 浏览

tensorflow - 在检查点中找不到 alexnet_v2/conv1/biases

我尝试保存和恢复alxene slim模型。但是当我运行代码时总是会出现这个错误saver.restore(sess, tf.train.latest_checkpoint('I:/model/mnist/'))。并引发错误: NotFoundError (see above for traceback): Key alexnet_v2/conv1/biases not found in checkpoint. 当我运行时tf.global_variables(),我只能得到conv2d的权重,结果没有偏差。我不明白问题是什么。这是我的代码:

这是我的亚历克斯模型

保存模型

恢复模型

0 投票
1 回答
1117 浏览

tensorflow - 更改优化器会导致“在检查点中找不到”错误

我已经使用rmsprop优化器为一定数量的步骤训练了 MobilenetV1 模型。现在我想将优化器从 切换rmspropadam并从这里开始恢复训练。但是,当它加载经过训练的模型时,当我将优化器从 更改为 时,会出现以下rmsprop错误adam。我正在使用 TF-slim 来训练模型。

谁能帮我解决这个问题?提前致谢。