我是机器学习领域的新手,刚刚起步。我一直在使用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数组文件并对其进行解析,但我不想这样做。)
答案 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