diff --git a/minigrida/loader/coord_tiles.py b/minigrida/loader/coord_tiles.py new file mode 100644 index 0000000..d24a4aa --- /dev/null +++ b/minigrida/loader/coord_tiles.py @@ -0,0 +1,36 @@ +#!/usr/bin/env python +# file coord_tiles.py +# author Florent Guiotte +# version 0.0 +# date 29 mai 2020 +"""Abstract + +doc. +""" + +from pathlib import Path +import rasterio as rio +import logging + +log = logging.getLogger() + + +def run(rasters_path, rasters_name, gt_suffix='gt.tif'): + gt_names = list(Path(rasters_path).glob('*' + gt_suffix)) + log.info('Found {} ground truth tiles.'.format(len(gt_names))) + + gt = [] + rasters = [] + + for gtn in gt_names: + print(gtn) + gt += [load_tif(gtn)] + + rasters += [{Path(n).stem: load_tif(gtn.as_posix().strip(gt_suffix) + n) + for n in rasters_name}] + + return gt, rasters + + +def load_tif(path): + return rio.open(str(path)).read()[0]