├── .gitignore ├── README.md ├── backbone └── mae.py ├── configs ├── _base_ │ ├── datasets │ │ ├── ade20k.py │ │ └── ade20k_640x640.py │ ├── default_runtime.py │ ├── models │ │ └── upernet.py │ └── schedules │ │ ├── schedule_160k.py │ │ └── schedule_320k.py └── mae │ └── upernet_mae_base_12_512_slide_160k_ade20k.py ├── log └── 20220131_012835.log ├── mmcv_custom ├── __init__.py ├── apex_runner │ ├── __init__.py │ ├── apex_iter_based_runner.py │ ├── checkpoint.py │ └── optimizer.py ├── checkpoint.py ├── layer_decay_optimizer_constructor.py ├── resize_transform.py └── train_api.py └── tools ├── dist_test.sh ├── dist_train.sh ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | *.py[cod] 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/README.md -------------------------------------------------------------------------------- /backbone/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/backbone/mae.py -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20k_640x640.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/configs/_base_/datasets/ade20k_640x640.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/configs/_base_/models/upernet.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_320k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/configs/_base_/schedules/schedule_320k.py -------------------------------------------------------------------------------- /configs/mae/upernet_mae_base_12_512_slide_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/configs/mae/upernet_mae_base_12_512_slide_160k_ade20k.py -------------------------------------------------------------------------------- /log/20220131_012835.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/log/20220131_012835.log -------------------------------------------------------------------------------- /mmcv_custom/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/mmcv_custom/__init__.py -------------------------------------------------------------------------------- /mmcv_custom/apex_runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/mmcv_custom/apex_runner/__init__.py -------------------------------------------------------------------------------- /mmcv_custom/apex_runner/apex_iter_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/mmcv_custom/apex_runner/apex_iter_based_runner.py -------------------------------------------------------------------------------- /mmcv_custom/apex_runner/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/mmcv_custom/apex_runner/checkpoint.py -------------------------------------------------------------------------------- /mmcv_custom/apex_runner/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/mmcv_custom/apex_runner/optimizer.py -------------------------------------------------------------------------------- /mmcv_custom/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/mmcv_custom/checkpoint.py -------------------------------------------------------------------------------- /mmcv_custom/layer_decay_optimizer_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/mmcv_custom/layer_decay_optimizer_constructor.py -------------------------------------------------------------------------------- /mmcv_custom/resize_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/mmcv_custom/resize_transform.py -------------------------------------------------------------------------------- /mmcv_custom/train_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/mmcv_custom/train_api.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/implus/mae_segmentation/HEAD/tools/train.py --------------------------------------------------------------------------------