Can you alter the label image too with STN?

Original article was published by /u/sleeplesscat91 on Deep Learning

I just know that there’s a method called Spatial Transformation Network for data augmentation. https://debuggercafe.com/spatial-transformer-network-using-pytorch/

If I train the data image with test images, given STN is just the spatial transformation, can I just transform the label image too using the same model?

submitted by /u/sleeplesscat91
[link] [comments]