你好作者，如果我希望使用我自己的灰度图来进行训练的话，除了yaml里inp_channels/out_channels，以及img_util.py、paired_image_dataset.py里的PairedImageDataset类、img2tensor、tensor2img、imfrombytes方法等，具体还有什么需要修改的地方吗？
你好作者，如果我希望使用我自己的灰度图来进行训练的话，除了yaml里inp_channels/out_channels，以及img_util.py、paired_image_dataset.py里的PairedImageDataset类、img2tensor、tensor2img、imfrombytes方法等，具体还有什么需要修改的地方吗？