├── .gitignore ├── LICENSE ├── README.md ├── base ├── __init__.py ├── base_data_loader.py ├── base_inference.py ├── base_model.py └── base_trainer.py ├── config ├── config_deeplab.json ├── config_fastscnn.json ├── config_hrnet.json ├── config_pspnet.json └── config_unet.json ├── data_loader ├── __init__.py ├── dataloader.py └── transforms.py ├── evaluation ├── __init__.py ├── losses.py └── metrics.py ├── models ├── __init__.py ├── backbones │ ├── __init__.py │ ├── mobilenetv2.py │ ├── resnet.py │ └── xception.py ├── deeplabv3_plus.py ├── fast_scnn.py ├── hrnet.py ├── pspnet.py └── unet.py ├── resource └── demod.mp4 ├── tools ├── bg.jpeg ├── video_infer.sh ├── video_infer_optical.sh ├── videoinfer.py └── videoinfer_opticalflow.py ├── train.py ├── train.sh ├── trainer └── trainer.py └── utils ├── __init__.py ├── flops_counter.py ├── logger.py ├── postprocess.py ├── utils.py └── visualization.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/README.md -------------------------------------------------------------------------------- /base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/base/__init__.py -------------------------------------------------------------------------------- /base/base_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/base/base_data_loader.py -------------------------------------------------------------------------------- /base/base_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/base/base_inference.py -------------------------------------------------------------------------------- /base/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/base/base_model.py -------------------------------------------------------------------------------- /base/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/base/base_trainer.py -------------------------------------------------------------------------------- /config/config_deeplab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/config/config_deeplab.json -------------------------------------------------------------------------------- /config/config_fastscnn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/config/config_fastscnn.json -------------------------------------------------------------------------------- /config/config_hrnet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/config/config_hrnet.json -------------------------------------------------------------------------------- /config/config_pspnet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/config/config_pspnet.json -------------------------------------------------------------------------------- /config/config_unet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/config/config_unet.json -------------------------------------------------------------------------------- /data_loader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_loader/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/data_loader/dataloader.py -------------------------------------------------------------------------------- /data_loader/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/data_loader/transforms.py -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluation/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/evaluation/losses.py -------------------------------------------------------------------------------- /evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/evaluation/metrics.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /models/backbones/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/models/backbones/mobilenetv2.py -------------------------------------------------------------------------------- /models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/models/backbones/resnet.py -------------------------------------------------------------------------------- /models/backbones/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/models/backbones/xception.py -------------------------------------------------------------------------------- /models/deeplabv3_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/models/deeplabv3_plus.py -------------------------------------------------------------------------------- /models/fast_scnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/models/fast_scnn.py -------------------------------------------------------------------------------- /models/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/models/hrnet.py -------------------------------------------------------------------------------- /models/pspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/models/pspnet.py -------------------------------------------------------------------------------- /models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/models/unet.py -------------------------------------------------------------------------------- /resource/demod.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/resource/demod.mp4 -------------------------------------------------------------------------------- /tools/bg.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/tools/bg.jpeg -------------------------------------------------------------------------------- /tools/video_infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/tools/video_infer.sh -------------------------------------------------------------------------------- /tools/video_infer_optical.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/tools/video_infer_optical.sh -------------------------------------------------------------------------------- /tools/videoinfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/tools/videoinfer.py -------------------------------------------------------------------------------- /tools/videoinfer_opticalflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/tools/videoinfer_opticalflow.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/train.sh -------------------------------------------------------------------------------- /trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/trainer/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/utils/flops_counter.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/utils/postprocess.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cavalleria/humanseg.pytorch/HEAD/utils/visualization.py --------------------------------------------------------------------------------