有哪些深度学习模型可以使用 RGB 通道以外的补充信息进行图像分割？

例如，想象一张拍摄效果不佳的河流（蓝色）图像显示了一个缺口，补充信息是详细的流向（箭头），这有助于显示河流的真实形状（现实中没有缺口）。为了得到河流的形状，我看到的大多数图像分割模型，比如 U-Net，只使用 RGB 通道。

是否有任何神经网络模型可以在图像分割任务的训练过程中使用这种辅助信息以及 RGB 通道？