├── .github └── workflows │ └── ci.yml ├── .gitignore ├── .pylintrc ├── .vscode ├── README.md ├── extensions.json └── settings.json ├── LICENSE ├── README.md ├── afa ├── __init__.py ├── config.py ├── data │ ├── __init__.py │ ├── build.py │ ├── callbacks │ │ ├── __init__.py │ │ ├── dumper.py │ │ └── evaluator.py │ ├── datasets │ │ ├── __init__.py │ │ ├── base.py │ │ ├── base_edge.py │ │ ├── bdd100k.py │ │ ├── bsds500.py │ │ ├── build.py │ │ ├── cityscapes.py │ │ ├── cityscapes_labels.py │ │ ├── custom.py │ │ ├── mapillary.py │ │ └── nyud.py │ ├── module.py │ ├── sampler.py │ ├── transforms │ │ ├── __init__.py │ │ ├── joint_transforms.py │ │ └── transforms.py │ └── uniform.py ├── model │ ├── __init__.py │ ├── afa_dla_ocr_seg_ssr.py │ ├── afa_dla_seg.py │ ├── afa_dla_seg_hma.py │ ├── afa_dla_seg_ssr.py │ ├── backbones │ │ ├── __init__.py │ │ └── dla.py │ ├── build.py │ ├── dla_seg.py │ ├── losses │ │ ├── __init__.py │ │ ├── build.py │ │ ├── cross_entropy.py │ │ └── rmi │ │ │ ├── __init__.py │ │ │ ├── rmi.py │ │ │ └── rmi_utils.py │ ├── module.py │ ├── necks │ │ ├── __init__.py │ │ ├── afa_dla_up.py │ │ ├── dla_up.py │ │ └── ocr.py │ ├── optimize │ │ ├── __init__.py │ │ ├── optimizer.py │ │ └── scheduler.py │ └── utils │ │ ├── __init__.py │ │ ├── attn_utils.py │ │ ├── dla_utils.py │ │ ├── mynn.py │ │ ├── ocr_utils.py │ │ ├── ops.py │ │ ├── seg_fix.py │ │ └── trnval_utils.py ├── trainer.py └── utils │ ├── __init__.py │ ├── logger.py │ ├── misc.py │ ├── opt.py │ └── structures.py ├── configs ├── bdd100k │ └── afa_dla_up_ocr_ssr_dla169.yaml ├── bsds │ ├── afa_dla_up_dla34.yaml │ └── afa_dla_up_dla34_pascal.yaml ├── cityscapes │ ├── afa_dla_up_ocr_ssr_dla102x.yaml │ └── afa_dla_up_ocr_ssr_dla102x_cv_3.yaml └── nyud │ ├── afa_dla_up_dla34.yaml │ └── afa_dla_up_dla34_hha.yaml ├── mypy.ini ├── pyproject.toml ├── readme ├── GETTING_STARTED.md ├── INSTALL.md ├── PREPARE_DATASETS.md └── src │ ├── Overview.png │ ├── bdd100k.png │ └── cityscapes.png ├── requirements.txt └── scripts └── lint.sh /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/.pylintrc -------------------------------------------------------------------------------- /.vscode/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/.vscode/README.md -------------------------------------------------------------------------------- /.vscode/extensions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/.vscode/extensions.json -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/README.md -------------------------------------------------------------------------------- /afa/__init__.py: -------------------------------------------------------------------------------- 1 | """AFA python interfaces and tools.""" 2 | -------------------------------------------------------------------------------- /afa/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/config.py -------------------------------------------------------------------------------- /afa/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/__init__.py -------------------------------------------------------------------------------- /afa/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/build.py -------------------------------------------------------------------------------- /afa/data/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/callbacks/__init__.py -------------------------------------------------------------------------------- /afa/data/callbacks/dumper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/callbacks/dumper.py -------------------------------------------------------------------------------- /afa/data/callbacks/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/callbacks/evaluator.py -------------------------------------------------------------------------------- /afa/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/__init__.py -------------------------------------------------------------------------------- /afa/data/datasets/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/base.py -------------------------------------------------------------------------------- /afa/data/datasets/base_edge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/base_edge.py -------------------------------------------------------------------------------- /afa/data/datasets/bdd100k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/bdd100k.py -------------------------------------------------------------------------------- /afa/data/datasets/bsds500.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/bsds500.py -------------------------------------------------------------------------------- /afa/data/datasets/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/build.py -------------------------------------------------------------------------------- /afa/data/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/cityscapes.py -------------------------------------------------------------------------------- /afa/data/datasets/cityscapes_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/cityscapes_labels.py -------------------------------------------------------------------------------- /afa/data/datasets/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/custom.py -------------------------------------------------------------------------------- /afa/data/datasets/mapillary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/mapillary.py -------------------------------------------------------------------------------- /afa/data/datasets/nyud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/datasets/nyud.py -------------------------------------------------------------------------------- /afa/data/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/module.py -------------------------------------------------------------------------------- /afa/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/sampler.py -------------------------------------------------------------------------------- /afa/data/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/transforms/__init__.py -------------------------------------------------------------------------------- /afa/data/transforms/joint_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/transforms/joint_transforms.py -------------------------------------------------------------------------------- /afa/data/transforms/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/transforms/transforms.py -------------------------------------------------------------------------------- /afa/data/uniform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/data/uniform.py -------------------------------------------------------------------------------- /afa/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/__init__.py -------------------------------------------------------------------------------- /afa/model/afa_dla_ocr_seg_ssr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/afa_dla_ocr_seg_ssr.py -------------------------------------------------------------------------------- /afa/model/afa_dla_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/afa_dla_seg.py -------------------------------------------------------------------------------- /afa/model/afa_dla_seg_hma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/afa_dla_seg_hma.py -------------------------------------------------------------------------------- /afa/model/afa_dla_seg_ssr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/afa_dla_seg_ssr.py -------------------------------------------------------------------------------- /afa/model/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | """AFA backbones init.""" 2 | -------------------------------------------------------------------------------- /afa/model/backbones/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/backbones/dla.py -------------------------------------------------------------------------------- /afa/model/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/build.py -------------------------------------------------------------------------------- /afa/model/dla_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/dla_seg.py -------------------------------------------------------------------------------- /afa/model/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/losses/__init__.py -------------------------------------------------------------------------------- /afa/model/losses/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/losses/build.py -------------------------------------------------------------------------------- /afa/model/losses/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/losses/cross_entropy.py -------------------------------------------------------------------------------- /afa/model/losses/rmi/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/losses/rmi/__init__.py -------------------------------------------------------------------------------- /afa/model/losses/rmi/rmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/losses/rmi/rmi.py -------------------------------------------------------------------------------- /afa/model/losses/rmi/rmi_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/losses/rmi/rmi_utils.py -------------------------------------------------------------------------------- /afa/model/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/module.py -------------------------------------------------------------------------------- /afa/model/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/necks/__init__.py -------------------------------------------------------------------------------- /afa/model/necks/afa_dla_up.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/necks/afa_dla_up.py -------------------------------------------------------------------------------- /afa/model/necks/dla_up.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/necks/dla_up.py -------------------------------------------------------------------------------- /afa/model/necks/ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/necks/ocr.py -------------------------------------------------------------------------------- /afa/model/optimize/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/optimize/__init__.py -------------------------------------------------------------------------------- /afa/model/optimize/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/optimize/optimizer.py -------------------------------------------------------------------------------- /afa/model/optimize/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/optimize/scheduler.py -------------------------------------------------------------------------------- /afa/model/utils/__init__.py: -------------------------------------------------------------------------------- 1 | """AFA network utils.""" 2 | -------------------------------------------------------------------------------- /afa/model/utils/attn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/utils/attn_utils.py -------------------------------------------------------------------------------- /afa/model/utils/dla_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/utils/dla_utils.py -------------------------------------------------------------------------------- /afa/model/utils/mynn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/utils/mynn.py -------------------------------------------------------------------------------- /afa/model/utils/ocr_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/utils/ocr_utils.py -------------------------------------------------------------------------------- /afa/model/utils/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/utils/ops.py -------------------------------------------------------------------------------- /afa/model/utils/seg_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/utils/seg_fix.py -------------------------------------------------------------------------------- /afa/model/utils/trnval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/model/utils/trnval_utils.py -------------------------------------------------------------------------------- /afa/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/trainer.py -------------------------------------------------------------------------------- /afa/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/utils/__init__.py -------------------------------------------------------------------------------- /afa/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/utils/logger.py -------------------------------------------------------------------------------- /afa/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/utils/misc.py -------------------------------------------------------------------------------- /afa/utils/opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/utils/opt.py -------------------------------------------------------------------------------- /afa/utils/structures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/afa/utils/structures.py -------------------------------------------------------------------------------- /configs/bdd100k/afa_dla_up_ocr_ssr_dla169.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/configs/bdd100k/afa_dla_up_ocr_ssr_dla169.yaml -------------------------------------------------------------------------------- /configs/bsds/afa_dla_up_dla34.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/configs/bsds/afa_dla_up_dla34.yaml -------------------------------------------------------------------------------- /configs/bsds/afa_dla_up_dla34_pascal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/configs/bsds/afa_dla_up_dla34_pascal.yaml -------------------------------------------------------------------------------- /configs/cityscapes/afa_dla_up_ocr_ssr_dla102x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/configs/cityscapes/afa_dla_up_ocr_ssr_dla102x.yaml -------------------------------------------------------------------------------- /configs/cityscapes/afa_dla_up_ocr_ssr_dla102x_cv_3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/configs/cityscapes/afa_dla_up_ocr_ssr_dla102x_cv_3.yaml -------------------------------------------------------------------------------- /configs/nyud/afa_dla_up_dla34.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/configs/nyud/afa_dla_up_dla34.yaml -------------------------------------------------------------------------------- /configs/nyud/afa_dla_up_dla34_hha.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/configs/nyud/afa_dla_up_dla34_hha.yaml -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/mypy.ini -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/pyproject.toml -------------------------------------------------------------------------------- /readme/GETTING_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/readme/GETTING_STARTED.md -------------------------------------------------------------------------------- /readme/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/readme/INSTALL.md -------------------------------------------------------------------------------- /readme/PREPARE_DATASETS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/readme/PREPARE_DATASETS.md -------------------------------------------------------------------------------- /readme/src/Overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/readme/src/Overview.png -------------------------------------------------------------------------------- /readme/src/bdd100k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/readme/src/bdd100k.png -------------------------------------------------------------------------------- /readme/src/cityscapes.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/readme/src/cityscapes.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/lint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/dla-afa/HEAD/scripts/lint.sh --------------------------------------------------------------------------------