你好我想处理灰度图像，这个单通道图像处理修改代码量如何

你好，我跑通过了这个代码，但是我发现我的mri图像是灰度的，出来的图像则是彩色的，不符合我的要求，现在我想要把这个通道设置为1了以后发现代码部分没有通道为1 的相关处理的代码，我想知道如果增加部分代码或修改部分代码能够实现单通道图片的处理么
