Почему tf.image.grayscale_to_rgb не работает с tenorflow 1.14.0? - PullRequest
Новая
       3

Почему tf.image.grayscale_to_rgb не работает с tenorflow 1.14.0?

0 голосов
/ 09 ноября 2019

Я пытаюсь преобразовать набор данных MNIST в RGB. Я использую следующий код, чтобы сделать это, но я получаю сообщение об ошибке, он работает с tenorflow1.12.0, но не работает с tenorflow1.14.0 (мой проект основан на tenorflow1.14.0). 

mnist = tf.keras.datasets.mnist
(x_train, _), (x_test, _) = mnist.load_data()

X = np.concatenate([x_train, x_test])
X = X.reshape((70000, 28, 28, 1))
assert X.shape == (70000, 28, 28, 1)

X = tf.image.grayscale_to_rgb(X, name=None)
assert X.shape == (70000, 28, 28, 3)

Я получаю ошибку: 

Traceback (most recent call last):
  File "PGE/redis_ako.py", line 75, in <module>
    X2 = tf.image.grayscale_to_rgb(X, name=None)
  File "/home/farshad/.virtualenvs/ako_cifar/local/lib/python2.7/site-packages/tensorflow/python/ops/image_ops_impl.py", line 1827, in grayscale_to_rgb
    images = _AssertGrayscaleImage(images)
  File "/home/farshad/.virtualenvs/ako_cifar/local/lib/python2.7/site-packages/tensorflow/python/ops/image_ops_impl.py", line 263, in _AssertGrayscaleImage
    _CheckGrayscaleImage(image, require_static=False), image)
  File "/home/farshad/.virtualenvs/ako_cifar/local/lib/python2.7/site-packages/tensorflow/python/ops/image_ops_impl.py", line 282, in _CheckGrayscaleImage
    if image.get_shape().ndims is None:
AttributeError: 'numpy.ndarray' object has no attribute 'get_shape'
...