search

ENSTA-U2IS-AI / torch-uncertainty

Open-source framework for uncertainty and deep learning models in PyTorch :seedling:
https://torch-uncertainty.github.io
Apache License 2.0
223 stars 16 forks source link

:sparkles: Add Superpixel-Mix augmentation for Segmentation models #87

Open o-laurent opened 3 months ago

o-laurent commented 3 months ago

Paper: https://arxiv.org/abs/2108.00968

Implementation: https://github.com/giannifranchi/deeplabv3-superpixelmix

More particularly, the transform is here: https://github.com/giannifranchi/deeplabv3-superpixelmix/blob/master/datasets/cityscapes_mix.py#L13