如何在张量流预训练模型中将输入输入到一层?

时间:2018-07-04 10:07:15

标签: tensorflow

预训练的模型有很多层,我想将输入直接输入到一个中间层(并丢弃前一层的结果)。 我只有那个模型的.pb文件和ckpt文件,那么如何在没有源代码的情况下修改计算流程?

这是我获得的唯一代码文件,但是我不知道如何使用它。图是由这个文件生成的吗?(与普通的tensorflow文件有很大不同)https://github.com/tensorflow/models/blob/master/research/object_detection/models/ssd_mobilenet_v2_feature_extractor.py

1 个答案:

答案 0 :(得分:0)

这是您需要做的:

  • 加载模型
  • 找到层的名称或检索要向其提供值的层的张量(为便于说明,将其命名为“ Z”)
  • 找到图层的名称或检索要从('Y')获取结果的图层的张量
  • 运行以下代码段:var str = "DeadSea"; var s = str.toLowerCase(); var arr = s.split(""); arr.forEach(a => { if (arr.indexOf(a) !== arr.lastIndexOf(a)) { str = str.replace(new RegExp(a, "gi"), ""); } }); console.log(str);