numpy
torch>=1.4
torchvision
pytorch_lightning
rasterio
scikit-image
scikit-learn
albumentations
einops
pandas
