torchange_example_changeos_swint_on_xview2_best42k

ChangeOS (Swin-T) — xView2 building damage assessment

ChangeOS with a Swin-T backbone, trained on xView2/xBD with torchange.

Scored with the official xView2 mixed metric, 0.3 * localization_F1 + 0.7 * harmonic_mean(damage_F1s), including the single-map constraint and the unlabelled-pixel (255) exclusion rule.

Results

Protocol: train on train+tier3 → select the checkpoint on test → report on hold. These weights are the test-selected checkpoint, evaluated once on hold.

Training recipe

Reproduction config and commands: examples/xview2_project/ in the torchange repo.

Usage

import torchange  # noqa: F401
from torchange.models.changeos import ChangeOS

model = ChangeOS.from_pretrained("EVER-Z/torchange_example_changeos_swint_on_xview2_best42k").eval()

# bitemporal input: pre- and post-event RGB stacked on the channel axis -> (B, 6, H, W)
# eval output: ChangeDetectionModelOutput with
#   change_prediction        (B, 5, H, W)  softmax damage probabilities
#   t1_semantic_prediction   (B, 1, H, W)  sigmoid building footprint
out = model(images)
loc = out['t1_semantic_prediction'] > 0.5
dam = loc * out['change_prediction'].argmax(dim=1)   # single-map constraint; needs B=1

Citation

@article{zheng2021changeos,
  title={Building damage assessment for rapid disaster response with a deep object-based semantic change detection framework: From natural disasters to man-made disasters},
  author={Zheng, Zhuo and Zhong, Yanfei and Wang, Junjue and Ma, Ailong and Zhang, Liangpei},
  journal={Remote Sensing of Environment},
  volume={265},
  pages={112636},
  year={2021},
  publisher={Elsevier}
}

@software{zheng2024torchange,
  author = {Zheng, Zhuo},
  title = {torchange: A Unified Change Representation Learning Benchmark Library},
  url = {https://github.com/Z-Zheng/pytorch-change-models},
  year = {2024}
}

@article{gupta2019xbd,
  title={xBD: A dataset for assessing building damage from satellite imagery},
  author={Gupta, Ritwik and Hosfelt, Richard and Sajeev, Sandra and Patel, Nirav and Goodman, Bryce and Doshi, Jigar and Heim, Eric and Choset, Howie and Gaston, Matthew},
  journal={arXiv preprint arXiv:1911.09296},
  year={2019}
}