├── .gitignore ├── LICENSE ├── README.md ├── Readme_ch.md ├── checkpoints └── .DS_Store ├── cityscapes.py ├── cityscapes_info.json ├── data ├── CamVid ├── gtFine └── leftImg8bit ├── evaluation.py ├── images ├── comparison-cityscapes.png ├── overview-of-our-method.png ├── stdc-architecture.png └── stdcseg-architecture.png ├── latentcy ├── .DS_Store ├── __init__.py ├── run_latency_stages.py └── utils │ ├── __init__.py │ ├── darts_utils.py │ └── genotypes.py ├── logger.py ├── logger.pyc ├── loss ├── detail_loss.py ├── loss.py └── util.py ├── models ├── __init__.py ├── model_stages.py └── model_stages_trt.py ├── modules ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-35.pyc │ ├── __init__.cpython-37.pyc │ ├── bn.cpython-35.pyc │ ├── bn.cpython-37.pyc │ ├── dense.cpython-35.pyc │ ├── dense.cpython-37.pyc │ ├── functions.cpython-35.pyc │ ├── functions.cpython-37.pyc │ ├── misc.cpython-35.pyc │ ├── misc.cpython-37.pyc │ ├── residual.cpython-35.pyc │ └── residual.cpython-37.pyc ├── bn.py ├── deeplab.py ├── dense.py ├── functions.py ├── misc.py ├── residual.py └── src │ ├── checks.h │ ├── inplace_abn.cpp │ ├── inplace_abn.h │ ├── inplace_abn_cpu.cpp │ ├── inplace_abn_cuda.cu │ ├── inplace_abn_cuda_half.cu │ └── utils │ ├── checks.h │ ├── common.h │ └── cuda.cuh ├── nets ├── __init__.py └── stdcnet.py ├── onnx.py ├── optimizer_loss.py ├── requirements.txt ├── scripts ├── cityscaptes_train_STDC1Seg.sh └── cityscaptes_train_STDC2Seg.sh ├── train.py ├── train.sh └── transform.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/README.md -------------------------------------------------------------------------------- /Readme_ch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/Readme_ch.md -------------------------------------------------------------------------------- /checkpoints/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/checkpoints/.DS_Store -------------------------------------------------------------------------------- /cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/cityscapes.py -------------------------------------------------------------------------------- /cityscapes_info.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/cityscapes_info.json -------------------------------------------------------------------------------- /data/CamVid: -------------------------------------------------------------------------------- 1 | /ssd/fanmingyuan/segmentation/data/CamVid -------------------------------------------------------------------------------- /data/gtFine: -------------------------------------------------------------------------------- 1 | /home/darknet/CM/data/Segmentation/cityScapes/gtFine -------------------------------------------------------------------------------- /data/leftImg8bit: -------------------------------------------------------------------------------- 1 | /home/darknet/CM/data/Segmentation/cityScapes/leftImg8bit -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/evaluation.py -------------------------------------------------------------------------------- /images/comparison-cityscapes.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/images/comparison-cityscapes.png -------------------------------------------------------------------------------- /images/overview-of-our-method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/images/overview-of-our-method.png -------------------------------------------------------------------------------- /images/stdc-architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/images/stdc-architecture.png -------------------------------------------------------------------------------- /images/stdcseg-architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/images/stdcseg-architecture.png -------------------------------------------------------------------------------- /latentcy/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/latentcy/.DS_Store -------------------------------------------------------------------------------- /latentcy/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /latentcy/run_latency_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/latentcy/run_latency_stages.py -------------------------------------------------------------------------------- /latentcy/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /latentcy/utils/darts_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/latentcy/utils/darts_utils.py -------------------------------------------------------------------------------- /latentcy/utils/genotypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/latentcy/utils/genotypes.py -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/logger.py -------------------------------------------------------------------------------- /logger.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/logger.pyc -------------------------------------------------------------------------------- /loss/detail_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/loss/detail_loss.py -------------------------------------------------------------------------------- /loss/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/loss/loss.py -------------------------------------------------------------------------------- /loss/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/loss/util.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/model_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/models/model_stages.py -------------------------------------------------------------------------------- /models/model_stages_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/models/model_stages_trt.py -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__init__.py -------------------------------------------------------------------------------- /modules/__pycache__/__init__.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/__init__.cpython-35.pyc -------------------------------------------------------------------------------- /modules/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /modules/__pycache__/bn.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/bn.cpython-35.pyc -------------------------------------------------------------------------------- /modules/__pycache__/bn.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/bn.cpython-37.pyc -------------------------------------------------------------------------------- /modules/__pycache__/dense.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/dense.cpython-35.pyc -------------------------------------------------------------------------------- /modules/__pycache__/dense.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/dense.cpython-37.pyc -------------------------------------------------------------------------------- /modules/__pycache__/functions.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/functions.cpython-35.pyc -------------------------------------------------------------------------------- /modules/__pycache__/functions.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/functions.cpython-37.pyc -------------------------------------------------------------------------------- /modules/__pycache__/misc.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/misc.cpython-35.pyc -------------------------------------------------------------------------------- /modules/__pycache__/misc.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/misc.cpython-37.pyc -------------------------------------------------------------------------------- /modules/__pycache__/residual.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/residual.cpython-35.pyc -------------------------------------------------------------------------------- /modules/__pycache__/residual.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/__pycache__/residual.cpython-37.pyc -------------------------------------------------------------------------------- /modules/bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/bn.py -------------------------------------------------------------------------------- /modules/deeplab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/deeplab.py -------------------------------------------------------------------------------- /modules/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/dense.py -------------------------------------------------------------------------------- /modules/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/functions.py -------------------------------------------------------------------------------- /modules/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/misc.py -------------------------------------------------------------------------------- /modules/residual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/residual.py -------------------------------------------------------------------------------- /modules/src/checks.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/src/checks.h -------------------------------------------------------------------------------- /modules/src/inplace_abn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/src/inplace_abn.cpp -------------------------------------------------------------------------------- /modules/src/inplace_abn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/src/inplace_abn.h -------------------------------------------------------------------------------- /modules/src/inplace_abn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/src/inplace_abn_cpu.cpp -------------------------------------------------------------------------------- /modules/src/inplace_abn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/src/inplace_abn_cuda.cu -------------------------------------------------------------------------------- /modules/src/inplace_abn_cuda_half.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/src/inplace_abn_cuda_half.cu -------------------------------------------------------------------------------- /modules/src/utils/checks.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/src/utils/checks.h -------------------------------------------------------------------------------- /modules/src/utils/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/src/utils/common.h -------------------------------------------------------------------------------- /modules/src/utils/cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/modules/src/utils/cuda.cuh -------------------------------------------------------------------------------- /nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nets/stdcnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/nets/stdcnet.py -------------------------------------------------------------------------------- /onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/onnx.py -------------------------------------------------------------------------------- /optimizer_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/optimizer_loss.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/cityscaptes_train_STDC1Seg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/scripts/cityscaptes_train_STDC1Seg.sh -------------------------------------------------------------------------------- /scripts/cityscaptes_train_STDC2Seg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/scripts/cityscaptes_train_STDC2Seg.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/train.sh -------------------------------------------------------------------------------- /transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/STDC-Seg/HEAD/transform.py --------------------------------------------------------------------------------