Как получить форму изображения после decode_jpeg в Tensorflow? - PullRequest
1 голос
/ 09 января 2020

У меня есть изображение, которое я передаю в tf.image.decode_jpeg:

img = tf.io.read_file(file_path)
img = tf.image.decode_jpeg(img, channels=3)

, и я пытаюсь получить его высоту и ширину img.shape[0] и img.shape[1], но оба возвращаются None. На самом деле, img.shape возвращает (None, None, 3).

Я использую это внутри функции, которая отображается в tf.data.Dataset. Как я могу получить реальную форму изображения?

обновление:

На данный момент я нашел решение, заключающееся в переносе кода на tf.py_function выполнить его с нетерпением, потому что набор данных создает внутренний граф. Я был бы признателен, если бы у кого-то было другое решение сделать это чисто графическим способом, который улучшил бы производительность.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...