Commit 52ce56c5 by Tristan Kreuziger

Update README.md

parent 9ab022d2
...@@ -33,8 +33,8 @@ After downloading the raw images from https://www.bigearth.net, they need to be ...@@ -33,8 +33,8 @@ After downloading the raw images from https://www.bigearth.net, they need to be
Run `python prep_splits.py -h` to see all available parameters. Run `python prep_splits.py -h` to see all available parameters.
## Using Pre-trained Models ## Pre-trained Models
The following models have been pre-trained on BigEarthNet: We provide code and models for fhe following models that have been pre-trained on BigEarthNet for scene classification:
| Model Name | Pre-Trained TensorFlow Model | | Model Name | Pre-Trained TensorFlow Model |
| ------------- |-------------------------------| | ------------- |-------------------------------|
...@@ -45,6 +45,8 @@ The following models have been pre-trained on BigEarthNet: ...@@ -45,6 +45,8 @@ The following models have been pre-trained on BigEarthNet:
| ResNet101 | [LINK]() | | ResNet101 | [LINK]() |
| ResNet152 | [LINK]() | | ResNet152 | [LINK]() |
The TensorFlow code for these models can be found [here](https://gitlab.tu-berlin.de/rsim/bigearthnet-models-tf).
## Training Own Models From Scratch ## Training Own Models From Scratch
Please check [`train.py`](https://gitlab.tu-berlin.de/rsim/bigearthnet-models-tf/blob/master/train.py) to see how the models can be trained from scratch with BigEarthNet. Please check [`train.py`](https://gitlab.tu-berlin.de/rsim/bigearthnet-models-tf/blob/master/train.py) to see how the models can be trained from scratch with BigEarthNet.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment