Input to `.fit()` should have rank 4. Got array with shape: (460, 512, 3)

问题遇到的现象和发生背景

问题相关代码，请勿粘贴截图

运行结果及报错内容 U-NET分割，keras中，Input to .fit() should have rank 4. Got array with shape: (460, 512, 3)怎么改呀，在网上看是reshape,请问是在在一部分对哪部分数据集呢？

我的解答思路和尝试过的方法

我想要达到的结果

试试下降输入的图片变成img=img[None],升维度，然后继续后面的步骤
不过你需要先确定下的模型时通道在前还是通道在后，也就是NCHW还是NHWC