AI修图输入维度不同

来源:6-8 编写训练神经网络的方法(上)

慕容2478845

2019-02-11

in train()
36 generated_images = g.predict(random_data, verbose = 0)
37
—> 38 input_batch = np.concatenate((input_batch, generated_images))
39
40 output_batch = [1] * BATCH_SIZE + [0] * BATCH_SIZE

ValueError: all the input arrays must have same number of dimensions

写回答

2回答

金华爱计算机呀

2019-04-08

请问你问题解决了么,可以分享一下么


0
0

Oscar

2019-03-07

给出的信息不够,我不能判断出错在哪里。我的原代码是没有错误的。

你上面贴出的三句代码我核对了我的代码,是一样的:

generated_images = g.predict(random_data, verbose=0)
input_batch = np.concatenate((input_batch, generated_images))
output_batch = [1] * BATCH_SIZE + [0] * BATCH_SIZE

错误应该来源于其他地方。你再对比下你的代码和我的代码,有没有什么其他不同之处。

0
2
羲和玄炎
回复
金华爱计算机呀
造成这个问题的原因是numpy 中的 concatenate 函数,输入需要的两个参数 axis 必须一样,也就是input_batch.shape 和 generated_images.shape 需要一致。 input_batch 是 从训练数据来的, 训练数据的取值参数有图像大小, 如果是 64 *64 那么不会报错,因为代码都是围绕着 这个数值进行的编码,其他的大小都会导致这个问题
2019-06-02
共2条回复

基于Python玩转人工智能最火框架 TensorFlow应用实践

机器学习入门,打牢TensorFlow框架应用是关键!

2214 学习 · 688 问题

查看课程