├── .gitignore ├── ENet_demo.gif ├── LICENSE ├── README.md ├── dataset ├── README.md ├── __init__.py ├── camvid.py ├── camvid │ ├── camvid_test_list.txt │ ├── camvid_train_list.txt │ ├── camvid_trainval_list.txt │ ├── camvid_val_list.txt │ ├── test │ │ └── .gitkeep │ ├── testannot │ │ └── .gitkeep │ ├── train │ │ └── .gitkeep │ ├── trainannot │ │ └── .gitkeep │ ├── val │ │ └── .gitkeep │ └── valannot │ │ └── .gitkeep ├── cityscape_scripts │ ├── __init__.py │ ├── download_cityscapes.sh │ ├── generate_mappings.py │ ├── print_utils.py │ └── process_cityscapes.py ├── cityscapes.py ├── cityscapes │ ├── cityscapes_test_list.txt │ ├── cityscapes_train_list.txt │ ├── cityscapes_trainval_list.txt │ ├── cityscapes_val_list.txt │ ├── gtCoarse │ │ ├── train │ │ │ └── .gitkeep │ │ ├── train_extra │ │ │ └── .gitkeep │ │ └── val │ │ │ └── .gitkeep │ ├── gtFine │ │ ├── test │ │ │ └── .gitkeep │ │ ├── train │ │ │ └── .gitkeep │ │ └── val │ │ │ └── .gitkeep │ └── leftImg8bit │ │ ├── test │ │ └── .gitkeep │ │ ├── train │ │ └── .gitkeep │ │ └── val │ │ └── .gitkeep ├── create_dataset_list.py └── inform │ ├── camvid_inform.pkl │ └── cityscapes_inform.pkl ├── docs ├── README.md ├── image-1.png └── requirements.yml ├── model ├── CGNet.py ├── ContextNet.py ├── DABNet.py ├── EDANet.py ├── ENet.py ├── ERFNet.py ├── ESNet.py ├── ESPNet.py ├── ESPNet_v2 │ ├── Model.py │ ├── SegmentationModel.py │ └── cnn_utils.py ├── FPENet.py ├── FSSNet.py ├── FastSCNN.py ├── LEDNet.py ├── LinkNet.py ├── SQNet.py ├── SegNet.py └── UNet.py ├── predict.py ├── requirements.txt ├── run_.sh ├── test.py ├── test_video.mp4 ├── test_video.py ├── tools ├── flops_counter │ ├── .gitignore │ ├── CHANGELOG.md │ ├── ENet_Flops_test.py │ ├── LICENSE │ ├── README.md │ ├── ptflops │ │ ├── __init__.py │ │ └── flops_counter.py │ ├── sample.py │ └── setup.py ├── fps_test │ └── eval_forward_time.py └── trainID2labelID.py ├── train.py └── utils ├── activations.py ├── colorize_mask.py ├── convert_state.py ├── debug.py ├── losses ├── __init__.py ├── loss.py └── lovasz_losses.py ├── metric ├── __init__.py └── metric.py ├── optim ├── AdamW.py ├── Lookahead.py ├── RAdam.py ├── Ranger.py └── __init__.py ├── scheduler ├── __init__.py └── lr_scheduler.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/.gitignore -------------------------------------------------------------------------------- /ENet_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/ENet_demo.gif -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/README.md -------------------------------------------------------------------------------- /dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/README.md -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/camvid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/camvid.py -------------------------------------------------------------------------------- /dataset/camvid/camvid_test_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/camvid/camvid_test_list.txt -------------------------------------------------------------------------------- /dataset/camvid/camvid_train_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/camvid/camvid_train_list.txt -------------------------------------------------------------------------------- /dataset/camvid/camvid_trainval_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/camvid/camvid_trainval_list.txt -------------------------------------------------------------------------------- /dataset/camvid/camvid_val_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/camvid/camvid_val_list.txt -------------------------------------------------------------------------------- /dataset/camvid/test/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/camvid/testannot/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/camvid/train/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/camvid/trainannot/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/camvid/val/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/camvid/valannot/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cityscape_scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cityscape_scripts/download_cityscapes.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/cityscape_scripts/download_cityscapes.sh -------------------------------------------------------------------------------- /dataset/cityscape_scripts/generate_mappings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/cityscape_scripts/generate_mappings.py -------------------------------------------------------------------------------- /dataset/cityscape_scripts/print_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/cityscape_scripts/print_utils.py -------------------------------------------------------------------------------- /dataset/cityscape_scripts/process_cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/cityscape_scripts/process_cityscapes.py -------------------------------------------------------------------------------- /dataset/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/cityscapes.py -------------------------------------------------------------------------------- /dataset/cityscapes/cityscapes_test_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/cityscapes/cityscapes_test_list.txt -------------------------------------------------------------------------------- /dataset/cityscapes/cityscapes_train_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/cityscapes/cityscapes_train_list.txt -------------------------------------------------------------------------------- /dataset/cityscapes/cityscapes_trainval_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/cityscapes/cityscapes_trainval_list.txt -------------------------------------------------------------------------------- /dataset/cityscapes/cityscapes_val_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/cityscapes/cityscapes_val_list.txt -------------------------------------------------------------------------------- /dataset/cityscapes/gtCoarse/train/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cityscapes/gtCoarse/train_extra/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cityscapes/gtCoarse/val/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cityscapes/gtFine/test/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cityscapes/gtFine/train/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cityscapes/gtFine/val/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cityscapes/leftImg8bit/test/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cityscapes/leftImg8bit/train/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cityscapes/leftImg8bit/val/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/create_dataset_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/create_dataset_list.py -------------------------------------------------------------------------------- /dataset/inform/camvid_inform.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/inform/camvid_inform.pkl -------------------------------------------------------------------------------- /dataset/inform/cityscapes_inform.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/dataset/inform/cityscapes_inform.pkl -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/image-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/docs/image-1.png -------------------------------------------------------------------------------- /docs/requirements.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/docs/requirements.yml -------------------------------------------------------------------------------- /model/CGNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/CGNet.py -------------------------------------------------------------------------------- /model/ContextNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/ContextNet.py -------------------------------------------------------------------------------- /model/DABNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/DABNet.py -------------------------------------------------------------------------------- /model/EDANet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/EDANet.py -------------------------------------------------------------------------------- /model/ENet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/ENet.py -------------------------------------------------------------------------------- /model/ERFNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/ERFNet.py -------------------------------------------------------------------------------- /model/ESNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/ESNet.py -------------------------------------------------------------------------------- /model/ESPNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/ESPNet.py -------------------------------------------------------------------------------- /model/ESPNet_v2/Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/ESPNet_v2/Model.py -------------------------------------------------------------------------------- /model/ESPNet_v2/SegmentationModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/ESPNet_v2/SegmentationModel.py -------------------------------------------------------------------------------- /model/ESPNet_v2/cnn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/ESPNet_v2/cnn_utils.py -------------------------------------------------------------------------------- /model/FPENet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/FPENet.py -------------------------------------------------------------------------------- /model/FSSNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/FSSNet.py -------------------------------------------------------------------------------- /model/FastSCNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/FastSCNN.py -------------------------------------------------------------------------------- /model/LEDNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/LEDNet.py -------------------------------------------------------------------------------- /model/LinkNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/LinkNet.py -------------------------------------------------------------------------------- /model/SQNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/SQNet.py -------------------------------------------------------------------------------- /model/SegNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/SegNet.py -------------------------------------------------------------------------------- /model/UNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/model/UNet.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/run_.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/test.py -------------------------------------------------------------------------------- /test_video.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/test_video.mp4 -------------------------------------------------------------------------------- /test_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/test_video.py -------------------------------------------------------------------------------- /tools/flops_counter/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/tools/flops_counter/.gitignore -------------------------------------------------------------------------------- /tools/flops_counter/CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/tools/flops_counter/CHANGELOG.md -------------------------------------------------------------------------------- /tools/flops_counter/ENet_Flops_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/tools/flops_counter/ENet_Flops_test.py -------------------------------------------------------------------------------- /tools/flops_counter/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/tools/flops_counter/LICENSE -------------------------------------------------------------------------------- /tools/flops_counter/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/tools/flops_counter/README.md -------------------------------------------------------------------------------- /tools/flops_counter/ptflops/__init__.py: -------------------------------------------------------------------------------- 1 | from .flops_counter import get_model_complexity_info 2 | -------------------------------------------------------------------------------- /tools/flops_counter/ptflops/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/tools/flops_counter/ptflops/flops_counter.py -------------------------------------------------------------------------------- /tools/flops_counter/sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/tools/flops_counter/sample.py -------------------------------------------------------------------------------- /tools/flops_counter/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/tools/flops_counter/setup.py -------------------------------------------------------------------------------- /tools/fps_test/eval_forward_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/tools/fps_test/eval_forward_time.py -------------------------------------------------------------------------------- /tools/trainID2labelID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/tools/trainID2labelID.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/train.py -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/colorize_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/colorize_mask.py -------------------------------------------------------------------------------- /utils/convert_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/convert_state.py -------------------------------------------------------------------------------- /utils/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/debug.py -------------------------------------------------------------------------------- /utils/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from .loss import * 2 | -------------------------------------------------------------------------------- /utils/losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/losses/loss.py -------------------------------------------------------------------------------- /utils/losses/lovasz_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/losses/lovasz_losses.py -------------------------------------------------------------------------------- /utils/metric/__init__.py: -------------------------------------------------------------------------------- 1 | from .metric import * 2 | -------------------------------------------------------------------------------- /utils/metric/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/metric/metric.py -------------------------------------------------------------------------------- /utils/optim/AdamW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/optim/AdamW.py -------------------------------------------------------------------------------- /utils/optim/Lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/optim/Lookahead.py -------------------------------------------------------------------------------- /utils/optim/RAdam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/optim/RAdam.py -------------------------------------------------------------------------------- /utils/optim/Ranger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/optim/Ranger.py -------------------------------------------------------------------------------- /utils/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/optim/__init__.py -------------------------------------------------------------------------------- /utils/scheduler/__init__.py: -------------------------------------------------------------------------------- 1 | from .lr_scheduler import * 2 | -------------------------------------------------------------------------------- /utils/scheduler/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/scheduler/lr_scheduler.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bharath5673/Efficient-Segmentation-Networks/HEAD/utils/utils.py --------------------------------------------------------------------------------