nielsr HF Staff commited on
Commit
30bdbc6
·
verified ·
1 Parent(s): 99054ab

Enhance model card: Add pipeline tag, library name, and sample usage

Browse files

This PR significantly improves the model card for UAGLNet by:
- Adding the `pipeline_tag: image-segmentation` to enhance discoverability on the Hugging Face Hub (e.g., at https://huggingface.co/models?pipeline_tag=image-segmentation).
- Specifying `library_name: pytorch`, based on evidence from the GitHub repository's installation instructions, which will enable an automated, predefined code snippet for PyTorch users.
- Including a direct link to the paper on Hugging Face: [https://huggingface.co/papers/2512.12941](https://huggingface.co/papers/2512.12941), alongside the existing arXiv link.
- Providing a concise description of the model based on the paper's abstract and GitHub README.
- Adding a "Sample Usage" section with code snippets directly from the GitHub README, demonstrating how to quickly reproduce results with pre-trained checkpoints.
- Removing the unnecessary "File information" section.

Please review and merge this PR if everything looks good.

Files changed (1) hide show
  1. README.md +20 -2
README.md CHANGED
@@ -1,9 +1,27 @@
1
  ---
2
  license: apache-2.0
 
 
3
  ---
4
 
5
  # UAGLNet
6
 
7
- **Repository:** [https://github.com/Dstate/UAGLNet](https://github.com/Dstate/UAGLNet)
8
 
9
- **Paper:** *“UAGLNet: Uncertainty-Aggregated Global-Local Fusion Network with Cooperative CNN-Transformer for Building Extraction”* ([arXiv:2512.12941](https://arxiv.org/abs/2512.12941))
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ pipeline_tag: image-segmentation
4
+ library_name: pytorch
5
  ---
6
 
7
  # UAGLNet
8
 
9
+ UAGLNet is an Uncertainty-Aggregated Global-Local Fusion Network designed for building extraction from remote sensing images. It exploits high-quality global-local visual semantics under the guidance of uncertainty modeling, addressing challenges posed by complex structural variations. The network features a novel cooperative encoder (hybrid CNN and transformer layers), an intermediate cooperative interaction block (CIB), a Global-Local Fusion (GLF) module, and an Uncertainty-Aggregated Decoder (UAD) to enhance segmentation accuracy by explicitly estimating pixel-wise uncertainty.
10
 
11
+ 📄 **Paper:** "[UAGLNet: Uncertainty-Aggregated Global-Local Fusion Network with Cooperative CNN-Transformer for Building Extraction](https://huggingface.co/papers/2512.12941)" ([arXiv:2512.12941](https://arxiv.org/abs/2512.12941))
12
+ 💻 **Repository:** [https://github.com/Dstate/UAGLNet](https://github.com/Dstate/UAGLNet)
13
+
14
+ ## Sample Usage
15
+
16
+ You can quickly reproduce the main results for various datasets by running `Reproduce.py`, which will load the pretrained checkpoints from Hugging Face and perform inference.
17
+
18
+ ```bash
19
+ # To reproduce results on the Inria dataset:
20
+ python Reproduce.py -d Inria
21
+
22
+ # To reproduce results on the Massachusetts dataset:
23
+ python Reproduce.py -d Mass
24
+
25
+ # To reproduce results on the WHU dataset:
26
+ python Reproduce.py -d WHU
27
+ ```