nok, could you provide more details as to your implementation, or a link to the notebook itself?
It’s hard to debug what’s going on without access to SegmentationMultiLabelList and open_mask.
I assume you saw florobax’s implementation here which I think is exactly what you’re trying to do as well?