Как правильно аугментировать маски Pytorch?

В задачах семантической сегментации, и пердсказания глубины возникает необходимость применить случайные пространственные аугментации как к изображению так и к маске. Но существуют так же pixel-level аугментации которые нужно применять только к изображениюю. Коректно сделать это можно используя библиотеку albumentation colab-notebook с кодом: