tensorflow.python.framework.errors_impl.InvalidArgumentError:不兼容的形状:[1572864]与[524288]

时间:2019-05-20 07:09:21

标签: python python-3.x tensorflow deep-learning

我对此错误有疑问。

tensorflow.python.framework.errors_impl.InvalidArgumentError:不兼容的形状:[1572864]与[524288]

因此它开始拟合模型,但由于错误而立即停止,并且无法找到问题所在。我确实找到了一些解决方案,说这在某些库中是一个问题,但是我认为该问题在代码中的某个位置,但是不确定我在哪里尝试了几件事,但没有解决方案。

整个错误在这里:     训练304个样本,验证92个样本     时代1/200 追溯(最近一次通话): 在train()中的第394行中的文件“ model.py” 火车validation_data =(imgs_valid,imgs_mask_valid)中的文件“ model.py”,第271行,回调= [model_checkpoint]) 文件“ /usr/local/lib/python3.5/dist-packages/keras/engine/training.py”,第1039行,适合验证(validation_steps = validation_steps) 文件“ /usr/local/lib/python3.5/dist-packages/keras/engine/training_arrays.py”,行199,在fit_loop outs = f(ins_batch)中 call 中的文件“ /usr/local/lib/python3.5/dist包/keras/backend/tensorflow_backend.py”,行2715返回self._call(输入) 在_call提取= self._callable_fn(* array_vals)中的文件“ /usr/local/lib/python3.5/dist包/keras/backend/tensorflow_backend.py”,第2675行 调用 run_metadata_ptr中的文件“ /usr/local/lib/python3.5/dist包/tensorflow/python/client/session.py”,第1439行) 在退出 c_api.TF_GetCode(self.status.status)中的文件“ /usr/local/lib/python3.5/dist包/tensorflow/python/framework/errors_impl.py”,第528行中tensorflow.python.framework.errors_impl.InvalidArgumentError:不兼容的形状:[1572864]与[524288]          [[{{node loss / conv2d_32_loss / mul}}]]          [[{{节点指标/ mean_iou / mean_iou_4 / confusion_matrix / ones_like}}]]

model.py file and preprocess function and get_dilated_unet function

在get_dilated_unet函数中,输入形状为ok图像为128x128x3,在预处理函数中,直到我注释了#imgs_p = imgs_p [...,np.newaxis]时,我确实犯了一个错误,即尺寸错误,尺寸为(304,128,128, 3,1),但应该是(304,128,128,3),但现在这个形状还可以,但是我再次收到此错误,因为形状不兼容并且不确定出了什么问题。

0 个答案:

没有答案