├── .gitignore ├── LICENSE ├── README.md ├── assests ├── Helvetica.ttf ├── ade │ ├── ADE_val_00000001.jpg │ └── ADE_val_00000049.jpg ├── banner.jpg ├── image_labels │ ├── Seq05VD_f05100.png │ └── Seq05VD_f05100_L.png └── infer_result.png ├── configs ├── ade20k.yaml ├── cityscapes.yaml ├── custom.yaml └── helen.yaml ├── docs ├── BACKBONES.md ├── DATASETS.md ├── MODELS.md └── OTHER_DATASETS.md ├── notebooks ├── aug_test.ipynb └── tutorial.ipynb ├── scripts ├── calc_class_weights.py ├── export_data.py ├── onnx_infer.py ├── openvino_infer.py ├── preprocess_celebamaskhq.py └── tflite_infer.py ├── semseg ├── __init__.py ├── augmentations.py ├── datasets │ ├── __init__.py │ ├── ade20k.py │ ├── atr.py │ ├── camvid.py │ ├── celebamaskhq.py │ ├── cihp.py │ ├── cityscapes.py │ ├── cocostuff.py │ ├── facesynthetics.py │ ├── helen.py │ ├── ibugmask.py │ ├── lapa.py │ ├── lip.py │ ├── mapillary.py │ ├── mhpv1.py │ ├── mhpv2.py │ ├── pascalcontext.py │ ├── suim.py │ └── sunrgbd.py ├── losses.py ├── metrics.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── convnext.py │ │ ├── micronet.py │ │ ├── mit.py │ │ ├── mobilenetv2.py │ │ ├── mobilenetv3.py │ │ ├── poolformer.py │ │ ├── pvt.py │ │ ├── resnet.py │ │ ├── resnetd.py │ │ ├── rest.py │ │ └── uniformer.py │ ├── base.py │ ├── bisenetv1.py │ ├── bisenetv2.py │ ├── custom_cnn.py │ ├── custom_vit.py │ ├── ddrnet.py │ ├── fchardnet.py │ ├── heads │ │ ├── __init__.py │ │ ├── condnet.py │ │ ├── fapn.py │ │ ├── fcn.py │ │ ├── fpn.py │ │ ├── lawin.py │ │ ├── segformer.py │ │ ├── sfnet.py │ │ └── upernet.py │ ├── lawin.py │ ├── layers │ │ ├── __init__.py │ │ ├── common.py │ │ └── initialize.py │ ├── modules │ │ ├── __init__.py │ │ ├── ppm.py │ │ └── psa.py │ ├── segformer.py │ └── sfnet.py ├── optimizers.py ├── schedulers.py └── utils │ ├── __init__.py │ ├── utils.py │ └── visualize.py ├── setup.py └── tools ├── benchmark.py ├── export.py ├── infer.py ├── train.py └── val.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/README.md -------------------------------------------------------------------------------- /assests/Helvetica.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/assests/Helvetica.ttf -------------------------------------------------------------------------------- /assests/ade/ADE_val_00000001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/assests/ade/ADE_val_00000001.jpg -------------------------------------------------------------------------------- /assests/ade/ADE_val_00000049.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/assests/ade/ADE_val_00000049.jpg -------------------------------------------------------------------------------- /assests/banner.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/assests/banner.jpg -------------------------------------------------------------------------------- /assests/image_labels/Seq05VD_f05100.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/assests/image_labels/Seq05VD_f05100.png -------------------------------------------------------------------------------- /assests/image_labels/Seq05VD_f05100_L.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/assests/image_labels/Seq05VD_f05100_L.png -------------------------------------------------------------------------------- /assests/infer_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/assests/infer_result.png -------------------------------------------------------------------------------- /configs/ade20k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/configs/ade20k.yaml -------------------------------------------------------------------------------- /configs/cityscapes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/configs/cityscapes.yaml -------------------------------------------------------------------------------- /configs/custom.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/configs/custom.yaml -------------------------------------------------------------------------------- /configs/helen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/configs/helen.yaml -------------------------------------------------------------------------------- /docs/BACKBONES.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/docs/BACKBONES.md -------------------------------------------------------------------------------- /docs/DATASETS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/docs/DATASETS.md -------------------------------------------------------------------------------- /docs/MODELS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/docs/MODELS.md -------------------------------------------------------------------------------- /docs/OTHER_DATASETS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/docs/OTHER_DATASETS.md -------------------------------------------------------------------------------- /notebooks/aug_test.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/notebooks/aug_test.ipynb -------------------------------------------------------------------------------- /notebooks/tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/notebooks/tutorial.ipynb -------------------------------------------------------------------------------- /scripts/calc_class_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/scripts/calc_class_weights.py -------------------------------------------------------------------------------- /scripts/export_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/scripts/export_data.py -------------------------------------------------------------------------------- /scripts/onnx_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/scripts/onnx_infer.py -------------------------------------------------------------------------------- /scripts/openvino_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/scripts/openvino_infer.py -------------------------------------------------------------------------------- /scripts/preprocess_celebamaskhq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/scripts/preprocess_celebamaskhq.py -------------------------------------------------------------------------------- /scripts/tflite_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/scripts/tflite_infer.py -------------------------------------------------------------------------------- /semseg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/__init__.py -------------------------------------------------------------------------------- /semseg/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/augmentations.py -------------------------------------------------------------------------------- /semseg/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/__init__.py -------------------------------------------------------------------------------- /semseg/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/ade20k.py -------------------------------------------------------------------------------- /semseg/datasets/atr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/atr.py -------------------------------------------------------------------------------- /semseg/datasets/camvid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/camvid.py -------------------------------------------------------------------------------- /semseg/datasets/celebamaskhq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/celebamaskhq.py -------------------------------------------------------------------------------- /semseg/datasets/cihp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/cihp.py -------------------------------------------------------------------------------- /semseg/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/cityscapes.py -------------------------------------------------------------------------------- /semseg/datasets/cocostuff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/cocostuff.py -------------------------------------------------------------------------------- /semseg/datasets/facesynthetics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/facesynthetics.py -------------------------------------------------------------------------------- /semseg/datasets/helen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/helen.py -------------------------------------------------------------------------------- /semseg/datasets/ibugmask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/ibugmask.py -------------------------------------------------------------------------------- /semseg/datasets/lapa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/lapa.py -------------------------------------------------------------------------------- /semseg/datasets/lip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/lip.py -------------------------------------------------------------------------------- /semseg/datasets/mapillary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/mapillary.py -------------------------------------------------------------------------------- /semseg/datasets/mhpv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/mhpv1.py -------------------------------------------------------------------------------- /semseg/datasets/mhpv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/mhpv2.py -------------------------------------------------------------------------------- /semseg/datasets/pascalcontext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/pascalcontext.py -------------------------------------------------------------------------------- /semseg/datasets/suim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/suim.py -------------------------------------------------------------------------------- /semseg/datasets/sunrgbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/datasets/sunrgbd.py -------------------------------------------------------------------------------- /semseg/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/losses.py -------------------------------------------------------------------------------- /semseg/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/metrics.py -------------------------------------------------------------------------------- /semseg/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/__init__.py -------------------------------------------------------------------------------- /semseg/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/__init__.py -------------------------------------------------------------------------------- /semseg/models/backbones/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/convnext.py -------------------------------------------------------------------------------- /semseg/models/backbones/micronet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/micronet.py -------------------------------------------------------------------------------- /semseg/models/backbones/mit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/mit.py -------------------------------------------------------------------------------- /semseg/models/backbones/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/mobilenetv2.py -------------------------------------------------------------------------------- /semseg/models/backbones/mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/mobilenetv3.py -------------------------------------------------------------------------------- /semseg/models/backbones/poolformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/poolformer.py -------------------------------------------------------------------------------- /semseg/models/backbones/pvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/pvt.py -------------------------------------------------------------------------------- /semseg/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/resnet.py -------------------------------------------------------------------------------- /semseg/models/backbones/resnetd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/resnetd.py -------------------------------------------------------------------------------- /semseg/models/backbones/rest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/rest.py -------------------------------------------------------------------------------- /semseg/models/backbones/uniformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/backbones/uniformer.py -------------------------------------------------------------------------------- /semseg/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/base.py -------------------------------------------------------------------------------- /semseg/models/bisenetv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/bisenetv1.py -------------------------------------------------------------------------------- /semseg/models/bisenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/bisenetv2.py -------------------------------------------------------------------------------- /semseg/models/custom_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/custom_cnn.py -------------------------------------------------------------------------------- /semseg/models/custom_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/custom_vit.py -------------------------------------------------------------------------------- /semseg/models/ddrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/ddrnet.py -------------------------------------------------------------------------------- /semseg/models/fchardnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/fchardnet.py -------------------------------------------------------------------------------- /semseg/models/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/heads/__init__.py -------------------------------------------------------------------------------- /semseg/models/heads/condnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/heads/condnet.py -------------------------------------------------------------------------------- /semseg/models/heads/fapn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/heads/fapn.py -------------------------------------------------------------------------------- /semseg/models/heads/fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/heads/fcn.py -------------------------------------------------------------------------------- /semseg/models/heads/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/heads/fpn.py -------------------------------------------------------------------------------- /semseg/models/heads/lawin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/heads/lawin.py -------------------------------------------------------------------------------- /semseg/models/heads/segformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/heads/segformer.py -------------------------------------------------------------------------------- /semseg/models/heads/sfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/heads/sfnet.py -------------------------------------------------------------------------------- /semseg/models/heads/upernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/heads/upernet.py -------------------------------------------------------------------------------- /semseg/models/lawin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/lawin.py -------------------------------------------------------------------------------- /semseg/models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/layers/__init__.py -------------------------------------------------------------------------------- /semseg/models/layers/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/layers/common.py -------------------------------------------------------------------------------- /semseg/models/layers/initialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/layers/initialize.py -------------------------------------------------------------------------------- /semseg/models/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/modules/__init__.py -------------------------------------------------------------------------------- /semseg/models/modules/ppm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/modules/ppm.py -------------------------------------------------------------------------------- /semseg/models/modules/psa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/modules/psa.py -------------------------------------------------------------------------------- /semseg/models/segformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/segformer.py -------------------------------------------------------------------------------- /semseg/models/sfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/models/sfnet.py -------------------------------------------------------------------------------- /semseg/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/optimizers.py -------------------------------------------------------------------------------- /semseg/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/schedulers.py -------------------------------------------------------------------------------- /semseg/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /semseg/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/utils/utils.py -------------------------------------------------------------------------------- /semseg/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/semseg/utils/visualize.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/setup.py -------------------------------------------------------------------------------- /tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/tools/benchmark.py -------------------------------------------------------------------------------- /tools/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/tools/export.py -------------------------------------------------------------------------------- /tools/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/tools/infer.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sithu31296/semantic-segmentation/HEAD/tools/val.py --------------------------------------------------------------------------------