golbin / TensorFlow-Tutorials

텐서플로우를 기초부터 응용까지 단계별로 연습할 수 있는 소스 코드를 제공합니다
2.13k stars 1k forks source link

안녕하세요 reshape 함수에 대해 도움을 주시면 감사하겠습니다. #33

Open smj21 opened 6 years ago

smj21 commented 6 years ago

올려주신 코드를 보고 연습을 하고 있는데 다음과 같은 문제가 생깁니다. 코드: for i in range(total_batch): batch_xs, batch_ys = data.next_batch(BatchSize=batch_size, shuffle=data_shuffle) batch_xs = np.asarray(batch_xs) batch_ys = np.asarray(batch_ys) batch_xs = batch_xs.reshape(-1, 48, 48, 1)

    _, cost_val = sess.run([optimizer, cost],
                           feed_dict={X: batch_xs,Y: batch_ys,keep_prob: 0.7})
    total_cost += cost_val
    data_shuffle = False

*batch size = 50입니다. X = tf.placeholder(tf.float32, [None, 48, 48, 1]) Y = tf.placeholder(tf.float32, [None, 7]) keep_prob = tf.placeholder(tf.float32)

이러식으로 돌렸을 때, 오류문이 다음과 같이 나옵니다. ValueError : cannot reshape array of size 50 into shape (48,48,1)

현재 파이썬3에서 위 코드를 돌리고 있는데, 버전이 달라서 reshape 함수가 안돌아가는 건가요?