torch
numpy
rasterio