在TensorFlow Lite中使用MobileNet v3进行对象检测

时间:2019-12-18 02:51:22

标签: android tensorflow object-detection tensorflow-lite mobilenet

一段时间以来,我一直在尝试使其在android设备上运行,但是我无法让模型进行任何检测。

我正在使用google提供的用于对象检测的示例代码:https://github.com/tensorflow/examples/tree/master/lite/examples/object_detection/android 并尝试使其与针对Mobilenet v3的经过预先训练的模型一起运行,尤其是从此处下载的“ ssd_mobilenet_v3_large_coco”中的.tflite文件:https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

每当我运行模型时,它总是为每个类别返回不大于10 ^ -15的置信度得分,即,它太低而无法构成检测结果。该代码在旧的mobilenet v1模型上运行良好,并且由于该模型是我唯一要更改的东西,因此我怀疑我一定在错误地使用新模型。

我想知道是否有人能够成功使用此新模型进行对象检测,如果可以,他们是如何做到的。这就是我现在要尝试的方式:

1。)采取google提供的上述示例代码,我已经将他们使用的模型替换为mobilenet v3 .tflite模型的.tflite文件。 (并相应地更改build.gradle文件)

2。)我已经设置了TF_OD_API_INPUT_SIZE = 320(在DetectorActivity.java中)

3。)我已经设置了NUM_DETECTIONS = 2034(在TFLiteObjectDetectionAPIModel.java中)

4。)我已将2个float数组用于检测输出,一个用于形状[1] [NUM_DETECTIONS] [91]的检测置信度,另一个用于形状[1] [NUM_DETECTIONS] [1] [ 4]

这些是essentailly我所做的唯一更改。为任何建议加油。

0 个答案:

没有答案