- 博客(1)
- 收藏
- 关注
原创 使用keras的predict(),报错:SlowBackpropInput: input and out_backprop must have the same batch size
写了一个做图像语义分割任务的自编码器,结构如下: 成功训练并且保存了模型,输入一张128*128的图像进行预测,如下: from tensorflow import keras # 预测 test_arr = test_arr.astype("float16")/255 pred_arr = (conv_autoencoder.predict(test_arr) > 0.5).astype("int32") 运行后报错:Conv2DSlowBackpropInput: input and
2021-08-09 19:15:18
445
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人