如何直接获取类名作为 yolo tf serving 的输出

Question

我想从我的自定义 yolov2 中获取输出作为已识别对象的字典列表。暗流有一个函数 return_predict() ，它做同样的事情，但在创建 tf 服务文件时。我无法提供我的输出层，这样我就会得到类似的结果。

代码链接：https://gist.github.com/sugartom/70b58505bf5f28d1cf5d05904f6c0af2

这里 tfnet.out - 给出一个张量，但我想要预测的类。

例如我想要的结果 - [{'label': 'buckle', 'confidence': 0.32669073, 'topleft': {'x': 368, 'y': 364}, 'bottomright': {'x '：420，'y'：414}}]

github 仓库：https://github.com/thtrieu/darkflow

Answer 1

我有同样的问题，但还没有找到方法。然而，预测张量的顺序总是相同的。因此，通过在开始时进行检查，结果可能与类有关。 对于我的发现，tf serving 不支持分类。除非对源代码进行了一些修改，否则只能预测有效。