改成rgb3通道输入遇到问题

[ZzWDlyy]

你好，李博，我尝试修改了你的代码改成了3通道的输入和输出，想得到彩色的图像，但是我跑出来的模型测试的结果非常的奇怪，这个模型是不是不适合rgb呢，还是说我需要更改损失函数什么的嘛？

[ZzWDlyy]

这是红外的图! 这是可见光的图 我使用两个相同的可见光融合也是全是红色背景，好像就只输出了R通道一样，我训练的把红外和可见光都放进去了。