.pbtx(labelmap)文件解析

时间:2018-06-21 11:10:55

标签: java android parsing tensorflow

我是机器学习领域的新手,刚刚起步。我一直在使用TensorFlow Mobile构建小型项目的电子元件检测器。 问题是如何在android中解析.pbtx labelmap,此文件格式看起来像JSON,但不是。

这是我的labelmap.pbtx文件的一部分

item {
  id: 1
  name: 'Resistor'
}
item {
  id: 2
  name: 'Zener Diode'
}
item {
  id: 3
  name: 'LED'
}

检测到后,thesorflow图会给出检测到的组件的ID,我想使用该组件的类ID来获取名称。

就像,如果id=3,那么我应该LED

Java中是否存在pbtx文件格式的解析器? (是的,我知道我可以为labelmap创建一个JSON数组文件并对其进行解析,但我不想这样做。)

1 个答案:

答案 0 :(得分:0)

1-训练完模型后,您应该生成一个具有以下扩展名的冻结图:.pb

-2由于您是新手,请使用tensorflow提供的android对象检测演示。

-3将冻结的图添加到演示中的assets文件夹中,并创建一个名为labels.txt的文件,其中包含.pbtxt文件的内容,如下所示:

???
Resistor
Zener Diode
LED

-4保存文件,并将其添加到资产文件夹中。

-5使用添加的新文件配置DetectorActivity。

-6现在,您可以检测到您的物体,然后做以后需要做的任何事情

演示链接:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/android

关于如何在您的android项目中使用冻结图的教程: https://towardsdatascience.com/detecting-pikachu-on-android-using-tensorflow-object-detection-15464c7a60cd