import tensorflow as tf

        labeled_poses = (labeled_rotations_3d, labeled_translations_3d,
                         labeled_sizes_3d)

        # Predictions
        predicted_classes = tf.cast(detections['detection_classes'], tf.int64)
        predicted_permutation = np.argsort(predicted_classes)
        predicted_classes = predicted_classes.numpy()[predicted_permutation]

