Update README.md
Browse files
README.md
CHANGED
|
@@ -22,37 +22,7 @@ UAGLNet addresses the challenges of building extraction from remote sensing imag
|
|
| 22 |
|
| 23 |
<img width="1000" src="https://github.com/Dstate/UAGLNet/raw/main/assets/architecture2.png">
|
| 24 |
|
| 25 |
-
##
|
| 26 |
-
|
| 27 |
-
### Installation
|
| 28 |
-
|
| 29 |
-
Clone this repository and create the environment.
|
| 30 |
-
```bash
|
| 31 |
-
git clone https://github.com/Dstate/UAGLNet.git
|
| 32 |
-
cd UAGLNet
|
| 33 |
-
|
| 34 |
-
conda create -n uaglnet python=3.8 -y
|
| 35 |
-
conda activate uaglnet
|
| 36 |
-
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
|
| 37 |
-
pip install -r requirements.txt
|
| 38 |
-
```
|
| 39 |
-
|
| 40 |
-
### Data Preprocessing
|
| 41 |
-
|
| 42 |
-
We conduct experiments on the Inria, WHU, and Massachusetts datasets. Detailed guidance for dataset preprocessing is provided here: [DATA_PREPARATION.md](https://github.com/Dstate/UAGLNet/blob/main/assets/DATA_PREPARATION.md).
|
| 43 |
-
|
| 44 |
-
### Training & Testing
|
| 45 |
-
|
| 46 |
-
Training and testing examples on the Inria dataset:
|
| 47 |
-
```bash
|
| 48 |
-
# training
|
| 49 |
-
python UAGLNet_train.py -c config/inria/UAGLNet.py
|
| 50 |
-
|
| 51 |
-
# testing
|
| 52 |
-
python UAGLNet_test.py -c config/inria/UAGLNet.py
|
| 53 |
-
```
|
| 54 |
-
|
| 55 |
-
### Main Results
|
| 56 |
|
| 57 |
The following table presents the performance of UAGLNet on building extraction benchmarks.
|
| 58 |
|
|
@@ -62,19 +32,6 @@ The following table presents the performance of UAGLNet on building extraction b
|
|
| 62 |
| Mass | 76.97 | 86.99 | 88.28 | 85.73 | [UAGLNet_Mass](https://huggingface.co/ldxxx/UAGLNet_Massachusetts) |
|
| 63 |
| WHU | 92.07 | 95.87 | 96.21 | 95.54 | [UAGLNet_WHU](https://huggingface.co/ldxxx/UAGLNet_WHU) |
|
| 64 |
|
| 65 |
-
You can quickly reproduce these results by running `Reproduce.py`, which will load the pretrained checkpoints from Hugging Face and perform inference.
|
| 66 |
-
|
| 67 |
-
```bash
|
| 68 |
-
# Inria
|
| 69 |
-
python Reproduce.py -d Inria
|
| 70 |
-
|
| 71 |
-
# Massachusetts
|
| 72 |
-
python Reproduce.py -d Mass
|
| 73 |
-
|
| 74 |
-
# WHU
|
| 75 |
-
python Reproduce.py -d WHU
|
| 76 |
-
```
|
| 77 |
-
|
| 78 |
## Citation
|
| 79 |
If you find this project useful in your research, please cite it as:
|
| 80 |
```bibtex
|
|
@@ -84,7 +41,4 @@ If you find this project useful in your research, please cite it as:
|
|
| 84 |
journal = {arXiv preprint arXiv:2512.12941},
|
| 85 |
year = {2025}
|
| 86 |
}
|
| 87 |
-
```
|
| 88 |
-
|
| 89 |
-
## Acknowledgement
|
| 90 |
-
This work is built upon [BuildingExtraction](https://github.com/stdcoutzrh/BuildingExtraction), [GeoSeg](https://github.com/WangLibo1995/GeoSeg/tree/main) and [SMT](https://github.com/AFeng-x/SMT). We sincerely appreciate their contributions which provide a clear pipeline and well-organized code.
|
|
|
|
| 22 |
|
| 23 |
<img width="1000" src="https://github.com/Dstate/UAGLNet/raw/main/assets/architecture2.png">
|
| 24 |
|
| 25 |
+
## Main Results
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
|
| 27 |
The following table presents the performance of UAGLNet on building extraction benchmarks.
|
| 28 |
|
|
|
|
| 32 |
| Mass | 76.97 | 86.99 | 88.28 | 85.73 | [UAGLNet_Mass](https://huggingface.co/ldxxx/UAGLNet_Massachusetts) |
|
| 33 |
| WHU | 92.07 | 95.87 | 96.21 | 95.54 | [UAGLNet_WHU](https://huggingface.co/ldxxx/UAGLNet_WHU) |
|
| 34 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
## Citation
|
| 36 |
If you find this project useful in your research, please cite it as:
|
| 37 |
```bibtex
|
|
|
|
| 41 |
journal = {arXiv preprint arXiv:2512.12941},
|
| 42 |
year = {2025}
|
| 43 |
}
|
| 44 |
+
```
|
|
|
|
|
|
|
|
|