pytorch
torchvision
pytorch_lightning
rasterio
