Remove torchvision

[cwerner]

Torchvision should not be necessary anymore since we use albumentations for data augmentation and should be removed.

[cwerner]

Seems like segmentation-models-pytorch depends on torchvision it so it'll still be pulled in...However, the only use of torchvision in our code is currently the ToTensor call which is also provided by albumentations.