Tensorflow对象检测api按图片中对象的顺序打印预测

Question

我使用tensorflow object-detection API训练了一个神经网络来解决简单的验证码问题，预测质量已经足够好了，但是当我使用以下代码输出预测时：

for index, value in enumerate(classes[0]):
    object_dict = {}
    if scores[0, index] > threshold:
        object_dict[(category_index.get(value)).get('name').encode('utf8')] = scores[0, index]
        objects.append(object_dict)

每次运行函数时，我都会得到随机顺序的预测结果。是否有可能按照对象在图片中的位置顺序获得预测的类？输出图片示例：

Picture

Answer 1

如果我理解这个问题，你想让它们从左到右排序吗？您需要获取方框坐标并对它们进行排序。

默认情况下，您将获得按分数排序的检测结果，从最高到最低，这就是您上面所做的。

获取boxes[0]的内容并定义从左到右的排序方法。你会只考虑盒子的xmin吗？xmax？两者的组合？有几种不同的方法可以选择这样做，但之后，您可以使用排序方法从左到右获得框的索引。