Open cwerner opened 2 years ago
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
.
Torchvision should not be necessary anymore since we use
albumentations
for data augmentation and should be removed.