Create databunch with multiple segmentation mask as label

[digitalspecialists (RobG)]

Have you seen this?
Sounds interesting, let us know how you get on.

Creating an Image Segmentation Dataset for Multiple Classes

You should use and ImageSegment with CHW channels. That probably requires a custom open method, so you will have to subclass ImageSegment but then the transforms should be applied properly to the targets.