我想从我的自定义yolov2获得输出，作为已识别对象的字典列表。Dark flow有一个函数return_predict()，它做同样的事情，但同时创建tf服务文件。我不能给我的输出层，这样我会得到类似的结果。
链接到代码：https://gist.github.com/sugartom/70b58505bf5f28d1cf5d05904f6c0af2
这里tfnet.out-给出了一个张量，但我想要预测类。
例如，我想要的结果-[｛'标签'："带扣"，"置信度"：0.32669073，"左上方"：｛'x'：368，"y"：364｝，"右下方"：｛
github回购：https://github.com/thtrieu/darkflow
我也有同样的问题，但还找不到方法。然而，预测张量的阶数总是相同的。因此，通过在开始时进行检查，结果可以与类相关。根据我的发现，tf serving不支持分类。除非对源代码进行了一些修改，否则只能预测工作。