Изменение формы изображения для графика вычисления тензорного потока - PullRequest
0 голосов
/ 03 июля 2019

Я пытаюсь построить граф вычислений впервые для классификации изображений.При подготовке / эксперименте для ввода я делаю эту предварительную обработку

image = tf.read_file('DSC_0367.JPG')
image = tf.image.decode_jpeg(image, channels=3)
image = tf.image.resize_images(image, [64, 64])

x = tf.placeholder(tf.float32, shape=(None, 12228))

with tf.Session() as sess:
    print(sess.run(tf.shape(image))) 

Я получаю форму изображения [64 64 3], насколько я понимаю, правильно, что перед подачей в граф вычислений мне нужно создать вектор данных входного изображения, которыйимеет размерность N x 1, где N 64 * 64 * 3 = 12228.

, если да и когда я пытаюсь изменить изображение с

image = tf.reshape(image, [12228]) 

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

ValueError: Невозможно изменить тензор с 12288 элементами для формирования [12228] (12228 элементов) для 'Reshape_12' (op: 'Reshape') с входными фигурами: [64,64,3], [1] ис входными тензорами, вычисленными как частичные фигуры: input [1] = [12228].

ниже этого я определяю свой скрытый слой 1 как tgis

w =  tf.Variable(tf.random_normal([12228, 100])
b =  tf.Variable(tf.random_normal([100])
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...