torch
torchvision
numpy
pillow
rasterio
tqdm