search

cwerner / deadtrees

Semantic Segmentation model for the detection of dead trees from ortho photos.
Apache License 2.0
9 stars 2 forks source link

Remove torchvision #74

Open cwerner opened 2 years ago

cwerner commented 2 years ago

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

cwerner commented 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.