numpy
rasterio
torch
