CANet

KO | EN

gitlite — search

#python #java #python3 #arduino #golang #machine-learning #rust #html #flask #javascript #seismology #nodejs

★ 10 Open GitHub ↗

Official implementation of RGB-D co-attention network published at PR.

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# CANet
CANet: Co-attention network for RGB-D semantic segmentation

## Citaion
You can cite our paper using: 
```
@article{zhou2022canet,
  title={CANet: Co-attention network for RGB-D semantic segmentation},
  author={Zhou, Hao and Qi, Lu and Huang, Hai and Yang, Xu and Wan, Zhaoliang and Wen, Xianglong},
  journal={Pattern Recognition},
  volume={124},
  pages={108468},
  year={2022},
  publisher={Elsevier}
}
```

## Table of Contents

- [Start](#Start)
- [Dependencies](#dependencies)
- [Usage](#usage)

## Start
Clone this repo:
```
git clone https://github.com/zhouhao94/CANet.git
```

## Dependencies
PyTorch 0.4.0, TensorboardX and other packages listed in `requirements.txt`:
```
torch==0.4.1
torchvision=0.2.2
numpy
imageio
scipy
tensorboardX
matplotlib
scikit-image
h5py
```

Install all dependent libraries:
```sh
$ pip install -r requirements.txt
```

## Usage

For training, you can pass the following argument:

```
python CANet_train_nyuv2.py --cuda -b 4
```

For inference, you should run the [CANet_eval_nyuv2.py](CANet_eval_nyuv2.py) like this:

```
python CANet_eval_nyuv2.py --cuda --last-ckpt /path/to/pretrained/model.pth
```

CANet

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README