Skip to content

Commit

Permalink
Merge pull request #72 from peterklipfel/issue_71-cropping-docs
Browse files Browse the repository at this point in the history
updated manual training docs to cover image cropping
  • Loading branch information
mhamilton723 authored Feb 2, 2023
2 parents e20df22 + 5c8abfd commit eb4d6b5
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -87,9 +87,10 @@ One can change the evaluation parameters and model by editing [`STEGO/src/config

## Training

To train STEGO from scratch, please first generate the KNN indices for the datasets of interest:
To train STEGO from scratch, please first generate the KNN indices for the datasets of interest. Note that this requires generating a cropped dataset first, and you may need to modify `crop datasets.py` to specify the dataset that you are cropping:

```shell script
python crop_datasets.py
python precompute_knns.py
```

Expand Down

0 comments on commit eb4d6b5

Please sign in to comment.