├── LICENSE ├── README.md ├── config ├── ade20k │ ├── ade20k_spnet101.yaml │ └── ade20k_spnet50.yaml └── cityscapes │ └── cityscapes_spnet101.yaml ├── data ├── ADE20K │ ├── train.txt │ └── val.txt └── Cityscapes │ ├── test.lst │ ├── train++.lst │ ├── train+.lst │ ├── train.lst │ ├── trainval.lst │ ├── val.lst │ └── val_test.lst ├── dataset ├── ade20k │ ├── ade20k_colors.txt │ └── ade20k_names.txt ├── bdd │ ├── bdddrivable_colors.txt │ ├── bdddrivable_names.txt │ ├── bddsemseg_colors.txt │ └── bddsemseg_names.txt ├── camvid │ ├── camvid_colors.txt │ └── camvid_names.txt ├── cityscapes │ ├── cityscapes_colors.txt │ └── cityscapes_names.txt └── voc2012 │ ├── voc2012_colors.txt │ └── voc2012_names.txt ├── lib ├── psa │ ├── functional.py │ ├── functions │ │ ├── __init__.py │ │ └── psamask.py │ ├── modules │ │ ├── __init__.py │ │ └── psamask.py │ └── src │ │ ├── __init__.py │ │ ├── cpu │ │ ├── operator.cpp │ │ ├── operator.h │ │ └── psamask.cpp │ │ └── gpu │ │ ├── operator.cpp │ │ ├── operator.h │ │ └── psamask_cuda.cu └── sync_bn │ ├── functions │ ├── __init__.py │ └── sync_bn.py │ ├── modules │ ├── __init__.py │ └── sync_bn.py │ └── src │ ├── __init__.py │ ├── cpu │ ├── operator.cpp │ ├── operator.h │ └── sync_bn.cpp │ └── gpu │ ├── common.h │ ├── device_tensor.h │ ├── operator.cpp │ ├── operator.h │ └── sync_bn_cuda.cu ├── models ├── __pycache__ │ ├── base.cpython-36.pyc │ ├── base.cpython-37.pyc │ ├── customize.cpython-36.pyc │ ├── customize.cpython-37.pyc │ ├── fcn.cpython-36.pyc │ ├── fcn.cpython-37.pyc │ ├── resnet.cpython-36.pyc │ ├── resnet.cpython-37.pyc │ ├── spnet.cpython-36.pyc │ └── spnet.cpython-37.pyc ├── base.py ├── customize.py ├── fcn.py ├── model_store.py ├── model_zoo.py ├── resnet.py └── spnet.py ├── run.sh ├── strip.png ├── tool ├── demo.py ├── test.py ├── test.sh ├── train.py └── train.sh └── util ├── __pycache__ ├── config.cpython-36.pyc ├── config.cpython-37.pyc ├── dataset.cpython-36.pyc ├── dataset.cpython-37.pyc ├── loss.cpython-36.pyc ├── transform.cpython-36.pyc ├── transform.cpython-37.pyc ├── util.cpython-36.pyc └── util.cpython-37.pyc ├── cityscapes.py ├── config.py ├── dataset.py ├── loss.py ├── transform.py └── util.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/README.md -------------------------------------------------------------------------------- /config/ade20k/ade20k_spnet101.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/config/ade20k/ade20k_spnet101.yaml -------------------------------------------------------------------------------- /config/ade20k/ade20k_spnet50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/config/ade20k/ade20k_spnet50.yaml -------------------------------------------------------------------------------- /config/cityscapes/cityscapes_spnet101.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/config/cityscapes/cityscapes_spnet101.yaml -------------------------------------------------------------------------------- /data/ADE20K/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/data/ADE20K/train.txt -------------------------------------------------------------------------------- /data/ADE20K/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/data/ADE20K/val.txt -------------------------------------------------------------------------------- /data/Cityscapes/test.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/data/Cityscapes/test.lst -------------------------------------------------------------------------------- /data/Cityscapes/train++.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/data/Cityscapes/train++.lst -------------------------------------------------------------------------------- /data/Cityscapes/train+.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/data/Cityscapes/train+.lst -------------------------------------------------------------------------------- /data/Cityscapes/train.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/data/Cityscapes/train.lst -------------------------------------------------------------------------------- /data/Cityscapes/trainval.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/data/Cityscapes/trainval.lst -------------------------------------------------------------------------------- /data/Cityscapes/val.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/data/Cityscapes/val.lst -------------------------------------------------------------------------------- /data/Cityscapes/val_test.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/data/Cityscapes/val_test.lst -------------------------------------------------------------------------------- /dataset/ade20k/ade20k_colors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/ade20k/ade20k_colors.txt -------------------------------------------------------------------------------- /dataset/ade20k/ade20k_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/ade20k/ade20k_names.txt -------------------------------------------------------------------------------- /dataset/bdd/bdddrivable_colors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/bdd/bdddrivable_colors.txt -------------------------------------------------------------------------------- /dataset/bdd/bdddrivable_names.txt: -------------------------------------------------------------------------------- 1 | void 2 | drivable 3 | alt 4 | -------------------------------------------------------------------------------- /dataset/bdd/bddsemseg_colors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/bdd/bddsemseg_colors.txt -------------------------------------------------------------------------------- /dataset/bdd/bddsemseg_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/bdd/bddsemseg_names.txt -------------------------------------------------------------------------------- /dataset/camvid/camvid_colors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/camvid/camvid_colors.txt -------------------------------------------------------------------------------- /dataset/camvid/camvid_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/camvid/camvid_names.txt -------------------------------------------------------------------------------- /dataset/cityscapes/cityscapes_colors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/cityscapes/cityscapes_colors.txt -------------------------------------------------------------------------------- /dataset/cityscapes/cityscapes_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/cityscapes/cityscapes_names.txt -------------------------------------------------------------------------------- /dataset/voc2012/voc2012_colors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/voc2012/voc2012_colors.txt -------------------------------------------------------------------------------- /dataset/voc2012/voc2012_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/dataset/voc2012/voc2012_names.txt -------------------------------------------------------------------------------- /lib/psa/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/psa/functional.py -------------------------------------------------------------------------------- /lib/psa/functions/__init__.py: -------------------------------------------------------------------------------- 1 | from .psamask import * 2 | -------------------------------------------------------------------------------- /lib/psa/functions/psamask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/psa/functions/psamask.py -------------------------------------------------------------------------------- /lib/psa/modules/__init__.py: -------------------------------------------------------------------------------- 1 | from .psamask import * 2 | -------------------------------------------------------------------------------- /lib/psa/modules/psamask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/psa/modules/psamask.py -------------------------------------------------------------------------------- /lib/psa/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/psa/src/__init__.py -------------------------------------------------------------------------------- /lib/psa/src/cpu/operator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/psa/src/cpu/operator.cpp -------------------------------------------------------------------------------- /lib/psa/src/cpu/operator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/psa/src/cpu/operator.h -------------------------------------------------------------------------------- /lib/psa/src/cpu/psamask.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/psa/src/cpu/psamask.cpp -------------------------------------------------------------------------------- /lib/psa/src/gpu/operator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/psa/src/gpu/operator.cpp -------------------------------------------------------------------------------- /lib/psa/src/gpu/operator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/psa/src/gpu/operator.h -------------------------------------------------------------------------------- /lib/psa/src/gpu/psamask_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/psa/src/gpu/psamask_cuda.cu -------------------------------------------------------------------------------- /lib/sync_bn/functions/__init__.py: -------------------------------------------------------------------------------- 1 | from .sync_bn import * 2 | -------------------------------------------------------------------------------- /lib/sync_bn/functions/sync_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/functions/sync_bn.py -------------------------------------------------------------------------------- /lib/sync_bn/modules/__init__.py: -------------------------------------------------------------------------------- 1 | from .sync_bn import * 2 | -------------------------------------------------------------------------------- /lib/sync_bn/modules/sync_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/modules/sync_bn.py -------------------------------------------------------------------------------- /lib/sync_bn/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/src/__init__.py -------------------------------------------------------------------------------- /lib/sync_bn/src/cpu/operator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/src/cpu/operator.cpp -------------------------------------------------------------------------------- /lib/sync_bn/src/cpu/operator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/src/cpu/operator.h -------------------------------------------------------------------------------- /lib/sync_bn/src/cpu/sync_bn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/src/cpu/sync_bn.cpp -------------------------------------------------------------------------------- /lib/sync_bn/src/gpu/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/src/gpu/common.h -------------------------------------------------------------------------------- /lib/sync_bn/src/gpu/device_tensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/src/gpu/device_tensor.h -------------------------------------------------------------------------------- /lib/sync_bn/src/gpu/operator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/src/gpu/operator.cpp -------------------------------------------------------------------------------- /lib/sync_bn/src/gpu/operator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/src/gpu/operator.h -------------------------------------------------------------------------------- /lib/sync_bn/src/gpu/sync_bn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/lib/sync_bn/src/gpu/sync_bn_cuda.cu -------------------------------------------------------------------------------- /models/__pycache__/base.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/__pycache__/base.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/base.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/__pycache__/base.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/customize.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/__pycache__/customize.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/customize.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/__pycache__/customize.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/fcn.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/__pycache__/fcn.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/fcn.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/__pycache__/fcn.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/__pycache__/resnet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/__pycache__/resnet.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/spnet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/__pycache__/spnet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/spnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/__pycache__/spnet.cpython-37.pyc -------------------------------------------------------------------------------- /models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/base.py -------------------------------------------------------------------------------- /models/customize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/customize.py -------------------------------------------------------------------------------- /models/fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/fcn.py -------------------------------------------------------------------------------- /models/model_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/model_store.py -------------------------------------------------------------------------------- /models/model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/model_zoo.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/spnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/models/spnet.py -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=0,1,2,3 sh tool/train.sh ade20k spnet50 2 | -------------------------------------------------------------------------------- /strip.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/strip.png -------------------------------------------------------------------------------- /tool/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/tool/demo.py -------------------------------------------------------------------------------- /tool/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/tool/test.py -------------------------------------------------------------------------------- /tool/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/tool/test.sh -------------------------------------------------------------------------------- /tool/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/tool/train.py -------------------------------------------------------------------------------- /tool/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/tool/train.sh -------------------------------------------------------------------------------- /util/__pycache__/config.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/__pycache__/config.cpython-36.pyc -------------------------------------------------------------------------------- /util/__pycache__/config.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/__pycache__/config.cpython-37.pyc -------------------------------------------------------------------------------- /util/__pycache__/dataset.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/__pycache__/dataset.cpython-36.pyc -------------------------------------------------------------------------------- /util/__pycache__/dataset.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/__pycache__/dataset.cpython-37.pyc -------------------------------------------------------------------------------- /util/__pycache__/loss.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/__pycache__/loss.cpython-36.pyc -------------------------------------------------------------------------------- /util/__pycache__/transform.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/__pycache__/transform.cpython-36.pyc -------------------------------------------------------------------------------- /util/__pycache__/transform.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/__pycache__/transform.cpython-37.pyc -------------------------------------------------------------------------------- /util/__pycache__/util.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/__pycache__/util.cpython-36.pyc -------------------------------------------------------------------------------- /util/__pycache__/util.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/__pycache__/util.cpython-37.pyc -------------------------------------------------------------------------------- /util/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/cityscapes.py -------------------------------------------------------------------------------- /util/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/config.py -------------------------------------------------------------------------------- /util/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/dataset.py -------------------------------------------------------------------------------- /util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/loss.py -------------------------------------------------------------------------------- /util/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/transform.py -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/houqb/SPNet/HEAD/util/util.py --------------------------------------------------------------------------------