README.md

Intra-inter Modal Attention Blocks for RGB-D Semantic Segmentation - Official Pytorch Implementation (ICMR 2023)

Intra-inter Modal Attention Blocks for RGB-D Semantic Segmentation

Soyun Choi, Youjia Zhang, Sungeun Hong

In Proceedings of the ACM International Conference on Multimedia Retrieval 2023

Setting

My env: RTX A6000, CUDA 11.2

Setting conda env:

conda create -n ima python=3.8
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
# install pytorch
pip install matplotlib numpy pandas opencv-python addict six albumentations==0.4.1 terminaltables pillow scipy h5py tensorboard

Datasets

You can download nyudv2 and sunrgbd dataset in here:

https://o365inha-my.sharepoint.com/:f:/g/personal/sychoi_office_inha_ac_kr/EieBA6oJ-JpLjxFhHKQ5D-sBEcoT4FI65EV9NjMVmmZ0HQ?e=2BICQq