laughtervv / Depthawarecnn
Licence: mit
Depth-aware CNN for RGB-D Segmentation, ECCV 2018
Stars: ✭ 247
Programming Languages
python
139335 projects - #7 most used programming language
Labels
Arxiv]
Depth-aware CNN for RGB-D Segmentation [Installation
Install Pytorch, dominate, TensorBoardX.
The depth-aware convolution and depth-aware average pooling operations are under folder models/ops
, to build them, simply use sh make.sh
to compile.
Training
#!./scripts/train.sh
python train.py \
--name nyuv2_VGGdeeplab_depthconv \
--dataset_mode nyuv2 \
--flip --scale --crop --colorjitter \
--depthconv \
--list dataset/lists/nyuv2/train.lst \
--vallist dataset/lists/nyuv2/val.lst
Pretrained Model can be found here.
Testing
#!./scripts/test.sh
python test.py \
--name nyuv2_VGGdeeplab_depthconv \
--dataset_mode nyuv2 \
--list dataset/lists/nyuv2/test.lst \
--depthconv \
--how_many 0
Citation
If you find this work useful, please consider citing:
@inproceedings{wang2018depthconv,
title={Depth-aware CNN for RGB-D Segmentation},
author={Wang, Weiyue and Neumann, Ulrich},
booktitle={ECCV},
year={2018}
}
Acknowledgemets
The visulization code is borrowed from pytorch-CycleGAN-and-pix2pix. Here is a pytorch implementation of DeepLab.
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].