Code base, tutorials and other information for the paper SolarMapper: estimating solar array location, size, and capacity using deep learning and overhead imagery.
- Edit hyper-parameters in
config.json
or make your custom config file with the same format asconfig.json
. - Run
python train.py --config CONFIG_PATH
if you're using your custom config file andCONFIG_PATH
is the path to your config file. Without the--config
flag,train.py
will look forconfig.json
in themrs
repo directory.
- Edit settings parameters (GPU ID, model path, etc.) in
evaluate.py
. Notice thatDS_NAME
should match the one in the config file of the trained model. - Run
python evaluate.py
.
- Edit settings parameters in
infer.py
in the same way as for model evaluation.DS_NAME
should match the one in the trained model, too. - The
FILE_LIST
parameter takes the path to a.txt
file which contains full paths of testing image files (one file path per row). - Run
python infer.py
.