UAV-based-Damaged-Buildings-Instance-Segmentation-model-DBA-RTMDet

This is based on RTMDet to construct identification of damaged buildings framework in post-earthquake UAV. All our code will be published later.

Train Test Inference

python tools/train_damaged.py

python tools/test_damaged.py

python demo/inference_damaged.py

Before executing the program, you need to modify the paths related to the model, data, etc. You can refer to sh/slurmtrain.sh and sh/slurminference.sh in the sh folder for details.

Dynamic Display of Results

Dynamic presentation of extraction results of damaged buildings in large-scale post-earthquake scenarios.

Detection - bounding box

Segmentation - imstance mask

Environment Require

Name Version Build Channel

libgccmutex 0.1 main
openmpmutex 5.1 1gnu
absl-py 2.1.0 py38h06a43080
addict 2.4.0 pypi0 pypi
aiohappyeyeballs 2.4.0 py38h06a43080
aiohttp 3.10.5 py38h5eee18b0
aiosignal 1.2.0 pyhd3eb1b00
aliyun-python-sdk-core 2.15.1 pypi0 pypi
aliyun-python-sdk-kms 2.16.4 pypi0 pypi
async-timeout 4.0.3 py38h06a43080
attrs 24.2.0 py38h06a43080
blas 1.0 mkl
blinker 1.6.2 py38h06a43080
brotli-python 1.0.9 py38h6a678d58
bzip2 1.0.8 h5eee18b6
c-ares 1.19.1 h5eee18b0
ca-certificates 2025.2.25 h06a43080
cachetools 5.3.3 py38h06a43080
certifi 2024.7.4 py38h06a43080
cffi 1.17.0 pypi0 pypi charset-normalizer 3.3.2 pyhd3eb1b00
click 8.1.7 py38h06a43080
colorama 0.4.6 pypi0 pypi contourpy 1.1.1 pypi0 pypi crcmod 1.7 pypi0 pypi cryptography 43.0.0 py38hdda00650
cuda-cudart 11.7.99 0 nvidia cuda-cupti 11.7.101 0 nvidia cuda-libraries 11.7.1 0 nvidia cuda-nvrtc 11.7.99 0 nvidia cuda-nvtx 11.7.91 0 nvidia cuda-runtime 11.7.1 0 nvidia cuda-version 12.6 3 nvidia cycler 0.12.1 pypi0 pypi ffmpeg 4.3 hf484d3e0 pytorch fightingcv-attention 1.0.0 pypi0 pypi filelock 3.14.0 pypi0 pypi fonttools 4.53.1 pypi0 pypi freetype 2.12.1 h4a9f2570
frozenlist 1.4.0 py38h5eee18b0
gmp 6.2.1 h295c9153
gmpy2 2.1.2 py38heeb90bb0
gnutls 3.6.15 he1e52480
google-auth 2.29.0 py38h06a43080
google-auth-oauthlib 0.5.2 py38h06a43080
grpcio 1.62.2 py38h6a678d50
idna 3.7 py38h06a43080
importlib-metadata 8.2.0 pypi0 pypi importlib-resources 6.4.0 pypi0 pypi intel-openmp 2023.1.0 hdb19cb546306
jinja2 3.1.4 py38h06a43080
jmespath 0.10.0 pypi0 pypi jpeg 9e h5eee18b3
kiwisolver 1.4.5 pypi0 pypi lame 3.100 h7b6447c0
lcms2 2.12 h3be64170
ldimpllinux-64 2.38 h11814591
lerc 3.0 h295c9150
libabseil 20240116.2 cxx17h6a678d50
libcublas 11.10.3.66 0 nvidia libcufft 10.7.2.124 h4fbf5900 nvidia libcufile 1.11.0.15 0 nvidia libcurand 10.3.7.37 0 nvidia libcusolver 11.4.0.1 0 nvidia libcusparse 11.7.4.91 0 nvidia libdeflate 1.17 h5eee18b1
libffi 3.4.4 h6a678d51
libgcc-ng 11.2.0 h12345671
libgomp 11.2.0 h12345671
libgrpc 1.62.2 h2d74bed0
libiconv 1.16 h5eee18b3
libidn2 2.3.4 h5eee18b0
libnpp 11.7.4.75 0 nvidia libnvjpeg 11.8.0.2 0 nvidia libpng 1.6.39 h5eee18b0
libprotobuf 4.25.3 he621ea30
libstdcxx-ng 11.2.0 h12345671
libtasn1 4.19.0 h5eee18b0
libtiff 4.5.1 h6a678d50
libunistring 0.9.10 h27cfd230
libwebp-base 1.3.2 h5eee18b0
lz4-c 1.9.4 h6a678d51
markdown 3.6 pypi0 pypi markdown-it-py 3.0.0 pypi0 pypi markupsafe 2.1.3 py38h5eee18b0
matplotlib 3.7.5 pypi0 pypi mdurl 0.1.2 pypi0 pypi mkl 2023.1.0 h213fc3f46344
mkl-service 2.4.0 py38h5eee18b1
mklfft 1.3.8 py38h5eee18b0
mklrandom 1.2.4 py38hdb19cb50
mmcv 2.1.0 pypi0 pypi mmdet 3.3.0 pypi0 pypi mmengine 0.10.4 pypi0 pypi model-index 0.1.11 pypi0 pypi mpc 1.1.0 h10f8cd91
mpfr 4.0.2 hb69a4c51
mpmath 1.3.0 py38h06a43080
multidict 6.0.4 py38h5eee18b0
ncurses 6.4 h6a678d50
nettle 3.7.3 hbbd107a1
networkx 3.1 py38h06a43080
numpy 1.24.3 py38hf6e82291
numpy-base 1.24.3 py38h060ed821
oauthlib 3.2.2 py38h06a43080
opencv-python 4.10.0.84 pypi0 pypi opendatalab 0.0.10 pypi0 pypi openh264 2.1.1 h4ff587b0
openjpeg 2.5.2 he7f1fd00
openmim 0.3.9 pypi0 pypi openssl 3.0.15 h5eee18b0
openxlab 0.1.1 pypi0 pypi ordered-set 4.1.0 pypi0 pypi oss2 2.17.0 pypi0 pypi packaging 24.1 pypi0 pypi pandas 2.0.3 pypi0 pypi pillow 10.4.0 py38h5eee18b0
pip 24.2 py38h06a43080
platformdirs 4.2.2 pypi0 pypi protobuf 4.25.3 py38h12ddb610
psutil 6.1.0 pypi0 pypi pyasn1 0.4.8 pyhd3eb1b00
pyasn1-modules 0.2.8 py0
pycocotools 2.0.7 pypi0 pypi pycparser 2.22 pypi0 pypi pycryptodome 3.20.0 pypi0 pypi pygments 2.18.0 pypi0 pypi pyjwt 2.8.0 py38h06a43080
pyopenssl 24.2.1 py38h06a43080
pyparsing 3.1.2 pypi0 pypi pysocks 1.7.1 py38h06a43080
python 3.8.19 h955ad1f0
python-dateutil 2.9.0.post0 pypi0 pypi pytorch 2.0.1 py3.8cuda11.7cudnn8.5.00 pytorch pytorch-cuda 11.7 h778d3585 pytorch pytorch-mutex 1.0 cuda pytorch pytz 2023.4 pypi0 pypi pyyaml 6.0.2 pypi0 pypi re2 2022.04.01 h295c9150
readline 8.2 h5eee18b0
requests 2.28.2 pypi0 pypi requests-oauthlib 2.0.0 py38h06a43080
rich 13.4.2 pypi0 pypi rsa 4.7.2 pyhd3eb1b01
scipy 1.10.1 pypi0 pypi setuptools 60.2.0 pypi0 pypi shapely 2.0.5 pypi0 pypi six 1.16.0 pypi0 pypi sqlite 3.45.3 h5eee18b0
sympy 1.12 py38h06a43080
tabulate 0.9.0 pypi0 pypi tbb 2021.8.0 hdb19cb50
tensorboard 2.12.1 py38h06a43080
tensorboard-data-server 0.7.0 py38h52d8a921
tensorboard-plugin-wit 1.8.1 py38h06a43080
termcolor 2.4.0 pypi0 pypi terminaltables 3.1.10 pypi0 pypi tk 8.6.14 h39e89690
tomli 2.0.1 pypi0 pypi torchaudio 2.0.2 py38cu117 pytorch torchtriton 2.0.0 py38 pytorch torchvision 0.15.2 py38cu117 pytorch tqdm 4.65.2 pypi0 pypi typingextensions 4.11.0 py38h06a43080
tzdata 2024.1 pypi0 pypi urllib3 1.26.19 pypi0 pypi werkzeug 3.0.3 py38h06a43080
wheel 0.43.0 py38h06a43080
xz 5.4.6 h5eee18b1
yapf 0.40.2 pypi0 pypi yarl 1.11.0 py38h5eee18b0
zipp 3.20.0 pypi0 pypi zlib 1.2.13 h5eee18b1
zstd 1.5.5 hc292b872