распечатать порядок классов в Object Detection API - PullRequest
0 голосов
/ 14 февраля 2019

Я использую Object_Detection_tutorial.ipynb.Я распечатал классы, найденные в API обнаружения объектов, но он печатает в случайном порядке.Как я могу распечатать его по заказу?Я предпочитаю сверху вниз изображения.

with detection_graph.as_default():
    with tf.Session(graph=detection_graph) as sess:
        for image_path in TEST_IMAGE_PATHS:
            img = Image.open(image_path)
            image = img.crop((0, 700, 500, 1600))
            image_np = load_image_into_numpy_array(image)
            image_np_expanded = np.expand_dims(image_np, axis=0)
            image_tensor = detection_graph.get_tensor_by_name('image_tensor:0')
            boxes = detection_graph.get_tensor_by_name('detection_boxes:0')
            scores = detection_graph.get_tensor_by_name('detection_scores:0')
            classes = detection_graph.get_tensor_by_name('detection_classes:0')
            num_detections = detection_graph.get_tensor_by_name('num_detections:0')

            (boxes, scores, classes, num_detections) = sess.run(
                [boxes, scores, classes, num_detections],
                feed_dict={image_tensor: image_np_expanded})

            vis_util.visualize_boxes_and_labels_on_image_array(
                image_np,
                np.squeeze(boxes),
                np.squeeze(classes).astype(np.int32),
                np.squeeze(scores),
                category_index,
                use_normalized_coordinates=True,
                line_thickness=5)

            print(image_np)
            objects = []
            threshold = 0.85
            for index, value in enumerate(classes[0]):
                object_dict = {}
                if scores[0, index] > threshold:
                    object_dict[(category_index.get(value)).get('name').encode('utf8')] = \
                    objects.append(object_dict)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...