Tensorflow对象检测api获取按边界框坐标排序的预测

问题描述 投票:0回答:1

我使用张量流对象检测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)

我在每次运行函数时都以随机顺序获得预测。之前我问了一个问题,建议我尝试使用坐标,但是找不到连接类和与此类关联的盒子的坐标的方法。随附了解决验证码的示例,因此我需要一种从左到右顺序输出预测的方法。

Image

python tensorflow computer-vision captcha object-detection-api
1个回答
0
投票
给出boxes[0]是一个形状为num_boxes * 4的数组,其中框中的第一个值为xmin,这可以为您提供框的索引,并按xmin最低的框(依次开始的那个)排序左)。
© www.soinside.com 2019 - 2024. All rights reserved.