利用多光谱数据集(tif)训练Resnet50,现在只能训练RGB图像,不知道怎么改

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

利用多光谱数据集(tif)训练Resnet50,现在只能训练RGB图像,不知道怎么改

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

img

运行结果及报错内容
我的解答思路和尝试过的方法
我想要达到的结果