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

Question

我想从我的自定义yolov2获取输出作为已识别对象的字典列表。黑暗流有一个函数return_predict（），它在创建服务文件时执行相同的操作。我无法给出输出层，以便得到类似的结果。

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

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

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

github repo：https://github.com/thtrieu/darkflow

Answer 1

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