！！Some questions about unsing my cdatasets

@CreamyLong 
大佬你好。想请教一下，关于在自己数据集上进行图像转换的实验。
我想基于你给出的分割掩码-真实图像转换的demo来进行RGB图像-RGB图像的转换。
在阅读您提供的代码中，我看到数据加载主要是stable-diffusion-master/ldm/data/custom_segmentation.py和stable-diffusion-master/ldm/data/seg_dataset.py两个文件。
但是您给出的demo中，掩码图像是灰度图，并且是单通道的。如果我的条件输入是3通道的RGB图像，我需要修改什么地方呢，是仅仅在seg_dataset.py文件中修改即可吗。
希望的到大佬您的指点。
不胜感激。