Maciej Sobkowiak
612f933e68
|
||
|---|---|---|
| assets | ||
| benchmark | ||
| data | ||
| models | ||
| research | ||
| src | ||
| .gitignore | ||
| README.md | ||
| main.ipynb | ||
| main.py | ||
| requirements.txt | ||
| setup.py | ||
README.md
Pobranie danych
python3 data/download_data.py --sas-url data/sas_westeurope.txt
Dane zostaną pobrane do dwóch folderów:
- data/train_features
- data/train_labels
Setup
Jeżeli pracujemy na systemie operacyjnym windows, paczki rosterio i GDAL należy pobrać z repozytorium link, a nastepnie zainstalować je lokalnie. Reszte paczek można zainstalować automatycznie komendą:
pip install -r requirements.txt
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
Jeżeli chcemy zwizualizować przykłady przekonwertowanych zdjęć, należy urochomić skrypt z flagą --show.
python3 setup.py --show
Training
Uruchomienie skryptu main urchomi proces kompilacji i trenowania modelu sieci UNET.
python3 main.py
Po wytrenowaniu, możliwe jest wygenerowanie przykładowych predykcji, a następne ich wyświetlenie, wywołując skrypt z flagą --show.
python3 main.py --predictions