├── Image ├── .md ├── Comparison-of-network-parameters.jpg ├── UMFormer.jpg ├── introduction.jpg ├── potsdam.jpg ├── uavid.jpg └── vaihingen.jpg ├── README.md ├── config ├── potsdam │ └── umformer.py ├── uavid │ └── umformer.py └── vaihingen │ └── umformer.py ├── geoseg ├── datasets │ ├── potsdam_dataset.py │ ├── transform.py │ ├── uavid_dataset.py │ └── vaihingen_dataset.py ├── losses │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── balanced_bce.cpython-39.pyc │ │ ├── bitempered_loss.cpython-39.pyc │ │ ├── dice.cpython-39.pyc │ │ ├── focal.cpython-39.pyc │ │ ├── focal_cosine.cpython-39.pyc │ │ ├── functional.cpython-39.pyc │ │ ├── jaccard.cpython-39.pyc │ │ ├── joint_loss.cpython-39.pyc │ │ ├── lovasz.cpython-39.pyc │ │ ├── soft_bce.cpython-39.pyc │ │ ├── soft_ce.cpython-39.pyc │ │ ├── soft_f1.cpython-39.pyc │ │ ├── useful_loss.cpython-39.pyc │ │ └── wing_loss.cpython-39.pyc │ ├── balanced_bce.py │ ├── bitempered_loss.py │ ├── cel1.py │ ├── dice.py │ ├── focal.py │ ├── focal_cosine.py │ ├── functional.py │ ├── jaccard.py │ ├── joint_loss.py │ ├── lovasz.py │ ├── soft_bce.py │ ├── soft_ce.py │ ├── soft_f1.py │ ├── useful_loss.py │ └── wing_loss.py └── models │ ├── My │ ├── IIA.py │ └── UMFormer.py │ └── umamba_v12 │ ├── __pycache__ │ ├── ssm.cpython-310.pyc │ ├── trans.cpython-310.pyc │ └── umamba_s.cpython-310.pyc │ ├── mamba │ ├── __pycache__ │ │ └── s7_mamba.cpython-310.pyc │ └── s7_mamba.py │ ├── ssm.py │ ├── trans.py │ └── umamba_s.py ├── inference_uavid.py ├── potsdam_test.py ├── requirements.txt ├── tools ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── cfg.cpython-39.pyc │ └── metric.cpython-39.pyc ├── cfg.py ├── metric.py ├── potsdam_patch_split.py ├── uavid_patch_split.py └── vaihingen_patch_split.py ├── train_supervision.py ├── uavid_test.py └── vaihingen_test.py /Image/.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Image/Comparison-of-network-parameters.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/Image/Comparison-of-network-parameters.jpg -------------------------------------------------------------------------------- /Image/UMFormer.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/Image/UMFormer.jpg -------------------------------------------------------------------------------- /Image/introduction.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/Image/introduction.jpg -------------------------------------------------------------------------------- /Image/potsdam.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/Image/potsdam.jpg -------------------------------------------------------------------------------- /Image/uavid.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/Image/uavid.jpg -------------------------------------------------------------------------------- /Image/vaihingen.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/Image/vaihingen.jpg -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/README.md -------------------------------------------------------------------------------- /config/potsdam/umformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/config/potsdam/umformer.py -------------------------------------------------------------------------------- /config/uavid/umformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/config/uavid/umformer.py -------------------------------------------------------------------------------- /config/vaihingen/umformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/config/vaihingen/umformer.py -------------------------------------------------------------------------------- /geoseg/datasets/potsdam_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/datasets/potsdam_dataset.py -------------------------------------------------------------------------------- /geoseg/datasets/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/datasets/transform.py -------------------------------------------------------------------------------- /geoseg/datasets/uavid_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/datasets/uavid_dataset.py -------------------------------------------------------------------------------- /geoseg/datasets/vaihingen_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/datasets/vaihingen_dataset.py -------------------------------------------------------------------------------- /geoseg/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__init__.py -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/balanced_bce.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/balanced_bce.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/bitempered_loss.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/bitempered_loss.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/dice.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/dice.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/focal.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/focal.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/focal_cosine.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/focal_cosine.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/functional.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/functional.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/jaccard.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/jaccard.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/joint_loss.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/joint_loss.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/lovasz.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/lovasz.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/soft_bce.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/soft_bce.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/soft_ce.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/soft_ce.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/soft_f1.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/soft_f1.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/useful_loss.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/useful_loss.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/__pycache__/wing_loss.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/__pycache__/wing_loss.cpython-39.pyc -------------------------------------------------------------------------------- /geoseg/losses/balanced_bce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/balanced_bce.py -------------------------------------------------------------------------------- /geoseg/losses/bitempered_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/bitempered_loss.py -------------------------------------------------------------------------------- /geoseg/losses/cel1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/cel1.py -------------------------------------------------------------------------------- /geoseg/losses/dice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/dice.py -------------------------------------------------------------------------------- /geoseg/losses/focal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/focal.py -------------------------------------------------------------------------------- /geoseg/losses/focal_cosine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/focal_cosine.py -------------------------------------------------------------------------------- /geoseg/losses/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/functional.py -------------------------------------------------------------------------------- /geoseg/losses/jaccard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/jaccard.py -------------------------------------------------------------------------------- /geoseg/losses/joint_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/joint_loss.py -------------------------------------------------------------------------------- /geoseg/losses/lovasz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/lovasz.py -------------------------------------------------------------------------------- /geoseg/losses/soft_bce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/soft_bce.py -------------------------------------------------------------------------------- /geoseg/losses/soft_ce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/soft_ce.py -------------------------------------------------------------------------------- /geoseg/losses/soft_f1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/soft_f1.py -------------------------------------------------------------------------------- /geoseg/losses/useful_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/useful_loss.py -------------------------------------------------------------------------------- /geoseg/losses/wing_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/losses/wing_loss.py -------------------------------------------------------------------------------- /geoseg/models/My/IIA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/models/My/IIA.py -------------------------------------------------------------------------------- /geoseg/models/My/UMFormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/models/My/UMFormer.py -------------------------------------------------------------------------------- /geoseg/models/umamba_v12/__pycache__/ssm.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/models/umamba_v12/__pycache__/ssm.cpython-310.pyc -------------------------------------------------------------------------------- /geoseg/models/umamba_v12/__pycache__/trans.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/models/umamba_v12/__pycache__/trans.cpython-310.pyc -------------------------------------------------------------------------------- /geoseg/models/umamba_v12/__pycache__/umamba_s.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/models/umamba_v12/__pycache__/umamba_s.cpython-310.pyc -------------------------------------------------------------------------------- /geoseg/models/umamba_v12/mamba/__pycache__/s7_mamba.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/models/umamba_v12/mamba/__pycache__/s7_mamba.cpython-310.pyc -------------------------------------------------------------------------------- /geoseg/models/umamba_v12/mamba/s7_mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/models/umamba_v12/mamba/s7_mamba.py -------------------------------------------------------------------------------- /geoseg/models/umamba_v12/ssm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/models/umamba_v12/ssm.py -------------------------------------------------------------------------------- /geoseg/models/umamba_v12/trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/models/umamba_v12/trans.py -------------------------------------------------------------------------------- /geoseg/models/umamba_v12/umamba_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/geoseg/models/umamba_v12/umamba_s.py -------------------------------------------------------------------------------- /inference_uavid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/inference_uavid.py -------------------------------------------------------------------------------- /potsdam_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/potsdam_test.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/tools/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /tools/__pycache__/cfg.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/tools/__pycache__/cfg.cpython-39.pyc -------------------------------------------------------------------------------- /tools/__pycache__/metric.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/tools/__pycache__/metric.cpython-39.pyc -------------------------------------------------------------------------------- /tools/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/tools/cfg.py -------------------------------------------------------------------------------- /tools/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/tools/metric.py -------------------------------------------------------------------------------- /tools/potsdam_patch_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/tools/potsdam_patch_split.py -------------------------------------------------------------------------------- /tools/uavid_patch_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/tools/uavid_patch_split.py -------------------------------------------------------------------------------- /tools/vaihingen_patch_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/tools/vaihingen_patch_split.py -------------------------------------------------------------------------------- /train_supervision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/train_supervision.py -------------------------------------------------------------------------------- /uavid_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/uavid_test.py -------------------------------------------------------------------------------- /vaihingen_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/takeyoutime/UMFormer/HEAD/vaihingen_test.py --------------------------------------------------------------------------------