Jonas1312 / pytorch-segmentation-dataset

SegmentationDataset class for torchvision. Applies data augmentation to both images and segmentations.