Выполните логический вывод с помощью пакета с моделью Tensorflow, имеющей в качестве входных данных tf_example - PullRequest
Новая
       4

Выполните логический вывод с помощью пакета с моделью Tensorflow, имеющей в качестве входных данных tf_example

0 голосов
/ 05 мая 2020

У меня есть модель, обученная с помощью Tensorflow 1.14, которая принимает в качестве входных данных изображение в форме tfexample: <tf.Tensor 'tf_example:0' shape=() dtype=string>.

Мне удалось передать ему одно изображение, создав tf_example таким образом : 

    _, buffer = cv2.imencode('.jpg', img)
    tf_example = tf.train.Example(features=tf.train.Features(feature={
            'image/encoded': tf.train.Feature(bytes_list=tf.train.BytesList(value=[buffer.tostring()])),
            'image/format': tf.train.Feature(bytes_list=tf.train.BytesList(value=[b'jpeg']))
    }))

Моя проблема в том, что я хотел бы выполнить логический вывод, используя пакет изображений, и я не знаю, как создать пакет, совместимый с вводом, ожидаемым сетью.

...