- Notifications
You must be signed in to change notification settings - Fork201
Gated-Shape CNN for Semantic Segmentation (ICCV 2019)
License
nv-tlabs/GSCNN
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
This is the official code for:
Towaki Takikawa,David Acuna,Varun Jampani,Sanja Fidler
ICCV 2019[Paper] [Project Page]
Based on based onhttps://github.com/NVIDIA/semantic-segmentation.
Copyright (C) 2019 NVIDIA Corporation. Towaki Takikawa, David Acuna, Varun Jampani, Sanja FidlerAll rights reserved.Licensed under the CC BY-NC-SA 4.0 license (https://creativecommons.org/licenses/by-nc-sa/4.0/legalcode).Permission to use, copy, modify, and distribute this software and its documentationfor any non-commercial purpose is hereby granted without fee, provided that the abovecopyright notice appear in all copies and that both that copyright notice and thispermission notice appear in supporting documentation, and that the name of the authornot be used in advertising or publicity pertaining to distribution of the softwarewithout specific, written prior permission.THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALLIMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR ANY PARTICULAR PURPOSE.IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIALDAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISINGOUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.~git clone https://github.com/nv-tlabs/GSCNNcd GSCNNCurrently, the code supports Python 3
- numpy
- PyTorch (>=1.1.0)
- torchvision
- scipy
- scikit-image
- tensorboardX
- tqdm
- torch-encoding
- opencv
- PyYAML
Download the pretrained model from theGoogle Drive Folder, and save it in 'checkpoints/'
Download (if needed) the inferred images from theGoogle Drive Folder
python train.py --evaluate --snapshot checkpoints/best_cityscapes_checkpoint.pth
A note on training- we train on 8 NVIDIA GPUs, and as such, training will be an issue with WiderResNet38 if you try to train on a single GPU.
If you use this code, please cite:
@article{takikawa2019gated, title={Gated-SCNN: Gated Shape CNNs for Semantic Segmentation}, author={Takikawa, Towaki and Acuna, David and Jampani, Varun and Fidler, Sanja}, journal={ICCV}, year={2019}}About
Gated-Shape CNN for Semantic Segmentation (ICCV 2019)
Topics
Resources
License
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Releases
Packages0
Contributors4
Uh oh!
There was an error while loading.Please reload this page.
