From 72852a7357e69cf0a672dc7b7e9ef7d5b2706d8f Mon Sep 17 00:00:00 2001 From: Maciej Sobkowiak Date: Wed, 16 Feb 2022 19:35:24 +0100 Subject: [PATCH] Prevent procesing converted images --- README.md | 10 +++------- src/setup.py | 4 +++- 2 files changed, 6 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index 436b85c..d785391 100644 --- a/README.md +++ b/README.md @@ -1,17 +1,13 @@ # Setup -W przypadku rosterio i GDAL należy pobrać wheele z repozytorium [tutaj](https://www.lfd.uci.edu/~gohlke/pythonlibs/), a nastepnie zainstalować je lokalnie. Reszta paczek zainstaluje się automatycznie komendą: +Jeżeli pracujemy na systemie operacyjnym windows, paczki rosterio i GDAL należy pobrać z repozytorium [link](https://www.lfd.uci.edu/~gohlke/pythonlibs/), a nastepnie zainstalować je lokalnie. Reszte paczek można zainstalować automatycznie komendą: ``` pip install -r requirements.txt ``` -Pierwsze uruchomienie polega na uruchomieniu skryptu setup.py +Pierwsze krok polega na uruchomieniu skryptu setup.py który sprawi że zdjęcia satelitarne zostają przekonwertowane z formatu tif na format jpeg. ``` python3 setup.py ``` -Skrypt ten sprawi że zdjęcia zostaną przekonwertowane z formatu tif na format jpg. - -#TODO - -Jeżeli chcemy zwizualizować przekonwertowane zdjęcia, należy urochomić skrypt z flagą --show +Jeżeli chcemy zwizualizować przykłady przekonwertowanych zdjęć, należy urochomić skrypt z flagą --show. ``` python3 setup.py --show ``` \ No newline at end of file diff --git a/src/setup.py b/src/setup.py index d2f7e4d..8db2fa4 100644 --- a/src/setup.py +++ b/src/setup.py @@ -22,7 +22,9 @@ if __name__ == "__main__": progress_bar(0, len(dirs), prefix = 'Converting TIF to JPG:', suffix = 'Complete', length = 50) for i, d in enumerate(dirs): progress_bar(i, len(dirs), prefix = 'Converting TIF to JPG:', suffix = 'Complete', length = 50) - convert_tif_to_jpg(os.path.join(FEATURES, d), os.path.join(LABELS, d + ".tif"), dp, fc, mk) + if(not os.path.exists(os.path.join(dp, d +'.jpeg')) or not os.path.exists(os.path.join(fc, d +'.jpeg')) or not os.path.exists(os.path.join(mk, d +'.jpeg'))): + convert_tif_to_jpg(os.path.join(FEATURES, d), os.path.join(LABELS, d + ".tif"), dp, fc, mk) elif(sys.argv[1] == '--show'): + img_names = [random.choice(os.listdir(dp)) for _ in range(3)] plot_image_grid(img_names) \ No newline at end of file