如何使用彩色蒙版图像将两幅图像之间的颜色转移过程引导到不同的区域？

Question

目前我有这个代码可以将颜色从一个图像转移到另一个图像：https://github.com/ProGamerGov/Neural-Tools/blob/master/linear-color-transfer.py

该代码的工作原理是将颜色从一个图像传输到另一个图像。

掩码图像的示例：https://i.imgur.com/jMQSl0f.png

相关的常规图片：https://i.imgur.com/Pl6WkL3.jpg

如何使用彩色蒙版图像将两幅图像之间的色彩转换过程引导到不同的区域？假设每个掩模中的区域数量可能不同，并且可以有不同的颜色，但只有掩模图像中颜色相同的区域才会进行颜色转移和应用。