Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
atalbanese committed Apr 10, 2023
1 parent a5d19fa commit 6e54b38
Showing 1 changed file with 24 additions and 1 deletion.
25 changes: 24 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,30 @@ python experiment_runner.py LOG_DIR RESULTS_FILE DATA_DIRECTORY EXPERIMENTS.CSV
| test_site | NEON site to test trained model on. Leave blank to test on same site. | Any four character NEON sitecode |
| pt_ckpt | Model checkpoint to load trained on unlabelled data. Only usable with deep learning. Generated using pretraining_run.py. Can be left blank. | A file location ending in .ckpt |

### Pre-Training
## Pre-Training
Optionally, users may train and then supply a deep learning model pre-trained on unlabelled NEON site data using a SWAV (SWapping Assignments between Views) model

python pretraining_run.py SAVEDIR DATA_DIRECTORY SITENAME

## Requirements

Full requirements incoming, but general requirements are:

- R:
- sp
- sf
- raster
- neonUtilities
- neonOS
- dplyr
- lidR

- Python:
- sklearn
- geopandas
- rasterio
- pytorch
- einops
- pytorch-lightning
- numpy
- scikit-image

0 comments on commit 6e54b38

Please sign in to comment.