Skip to content

Latest commit

 

History

History
82 lines (58 loc) · 2.45 KB

README.md

File metadata and controls

82 lines (58 loc) · 2.45 KB
Raw

Notes

  • For mapping center pivot irrigation systems, please, see our latest work here.

Get gdal development libraries:

$ sudo apt-add-repository ppa:ubuntugis/ubuntugis-unstable
$ sudo apt-get update
$ sudo apt-get install libgdal-dev
$ sudo apt-get install python3-dev
$ sudo apt-get install gdal-bin python3-gdal

Create and activate a virtual environment:

$ virtualenv env -p python3
$ source env/bin/activate

Install GDAL:

(env) $ pip3 install numpy
(env) $ pip3 install GDAL==$(gdal-config --version) --global-option=build_ext --global-option="-I/usr/include/gdal"

Install TensorFlow-GPU

(env) $ pip3 install tensorflow-gpu

Install Others Requirements

(env) $ pip3 install -r requirements.txt

Build datasets

Build dataset to train

python3 run.py --mode=generate --image=image1.tif --labels=image1_labels.tif --output=train.h5 --chip_size=512 --channels=4 --grids=2 --rotate=true
python3 run.py --mode=generate --image=image2.tif --labels=image2_labels.tif --output=train.h5 --chip_size=512 --channels=4 --grids=2 --rotate=true

Build dataset to test

python3 run.py --mode=generate --image=image3.tif --labels=image3_labels.tif --output=test.h5 --chip_size=512 --channels=4 --grids=2 --rotate=true
python3 run.py --mode=generate --image=image4.tif --labels=image4_labels.tif --output=test.h5 --chip_size=512 --channels=4 --grids=2 --rotate=true

Build dataset to validation

python3 run.py --mode=generate --image=image5.tif --labels=image5_labels.tif --output=validation.h5 --chip_size=512 --channels=4 --grids=2 --rotate=true

Train model

python3 run.py --mode=train --train=train.h5 --test=test.h5 --epochs=100 --batch_size=5 --classes=2

The trained model is available here: https://drive.google.com/file/d/11RO6vJL6eYmtz2YlsEGmz2hUPJ3H1rqd/view?usp=sharing

Evaluate model

python3 run.py --mode=evaluate --evaluate=validation.h5 --batch_size=5 --classes=2

Predict image

python3 run.py --mode=predict --input=image.tif --output=output.tif --chip_size=1024 --channels=4 --grids=1 --batch_size=5 --classes=2

Paper

Saraiva, M.; Protas, É.; Salgado, M.; Souza, C. Automatic Mapping of Center Pivot Irrigation Systems from Satellite Images Using Deep Learning. Remote Sens. 2020, 12, 558. DOI: 10.3390/rs12030558