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

Question

我训练了神经网络以使用tensorflowobject-detectionAPI解决简单的验证码，预测质量足够好，但是当我使用以下代码输出预测时：

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？两者结合？您可以选择几种不同的方法来执行此操作，但是之后，可以使用排序方法从左到右获取框的索引。

Get live help from me