gpt4 book ai didi

android - 从 TensorFlow Android Camera Demo 重新训练 Inception5h 模型

转载 作者:塔克拉玛干 更新时间:2023-11-03 00:19:42 24 4
gpt4 key购买 nike

TensorFlow Android Camera Demo使用 Inception5h model用于提供卓越性能的实时图像识别。由于我没有成功地重新训练 Inception5h,所以我选择了 InceptionV3 model但它在图像识别方面并不那么活泼。所以我回到开始尝试重新训练(或迁移学习)Inception5h 模型。我试过修改 retrain.py但它显然只是为 v3 模型编写的。 5h 模型不包含“pool_3/_reshape:0”、“DecodeJpeg/contents:0”或“ResizeBilinear:0”张量。还有其他差异。

我在机器学习和 TensorFlow 方面有点新手,所以我非常感谢关于我必须做什么的明确步骤。

谢谢!

最佳答案

它看起来像 retrain.py脚本和 tutorial刚刚更新以使用 mobilenet架构。

这样就解决了你问题的第一部分,它实际上不是 inception5h,但它在移动设备上运行良好,准确性比 inception5h 好得多。

要真正让它在 android 示例中运行,您仍然需要更新 these settings .

我想你应该可以只复制 the settings determined for the mobilenet you choose, from the retrain script你可能没事。


如果您想使用不同的网络,而该网络在 retrain.py 中没有设置,那么我能想到的确定它们的最简单方法是使用 TensorBoard 探索图表。

因此,如果您真的想要使用 inception 5h,您可以下载并解压它:

curl -O https://storage.googleapis.com/download.tensorflow.org/models/inception5h.zip
unzip -d inception5h inception5h.zip

然后从 Tensorflow for Poets: 2 codelab 中获取这个简单的脚本repo , 将图形 .pb 文件转换为 tensorboard 可以使用的东西:

curl -O https://raw.githubusercontent.com/googlecodelabs/tensorflow-for-poets-2/master/scripts/graph_pb2tb.py

然后在你的 graph.pb 上运行它:

mkdir tb_graph
python graph_pb2tb.py tb/inception5h inception5h/tensorflow_inception_graph.pb

open it in tensorboard :

tensorboard --logdir tb_graph

然后在图中四处寻找并找到您需要填充自己的节点的名称可能相对简单 model_info听写。

我认为这是您要设置为bottleneck_tensor 的节点:

TensorBoard screenshot of inception 5h with avgpool0/reshape highlighted

关于android - 从 TensorFlow Android Camera Demo 重新训练 Inception5h 模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42003846/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com