请问为什么image_shape只能是[3, 512, 512]?

[sidongfeng]

报错原因：Error: ShapeError: broadcast dimension mismatch

[dyning]

这个是模型训练，还是预测？

[sidongfeng]

训练模型

[LDOUBLEV]

可否提供完整的报错信息和运行指令

[sidongfeng]

运行指令：

报错信息：

[LDOUBLEV]

已修复，可以更新下代码

[sidongfeng]

谢谢更新。

[BruceHan98]

你好，我在det_db预训练模型上训练自己的数据集（图片尺寸不是默认的（3，640，640））时，修改相应的配置文件的image_shape为自己数据集的图片尺寸，出现了同样的情况：Broadcast dimension mismatch.
请问需要修改哪些文件来解决？

Error Message Summary:

InvalidArgumentError: Broadcast dimension mismatch. Operands could not be broadcast together with the shape of X = [16, 96, 64, 10] and the shape of Y = [16, 96, 63, 10]. Received [64] in X is not equal to [63] in Y at i:2.