У меня есть изображение, которое я передаю в 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
выполнить его с нетерпением, потому что набор данных создает внутренний граф. Я был бы признателен, если бы у кого-то было другое решение сделать это чисто графическим способом, который улучшил бы производительность.