├── .gitignore ├── LICENSE ├── README.md ├── black.toml ├── compress.py ├── cut_test_dataset.py ├── fit_predict.py ├── fit_predict2.py ├── inria ├── augmentations.py ├── dataset.py ├── factory.py ├── losses.py ├── metric.py ├── models │ ├── __init__.py │ ├── can.py │ ├── deeplab.py │ ├── efficient_unet.py │ ├── fpn.py │ ├── hg.py │ ├── hrnet.py │ ├── u2net.py │ └── unet.py ├── optim.py ├── pseudo.py ├── scheduler.py └── visualization.py ├── make_train_tiles.py ├── requirements.txt ├── run_tensorboard.cmd ├── run_tensorboard.sh ├── sample_color.jpg ├── scripts ├── train.cmd ├── train_b6.sh ├── train_hrnet18_4x1080Ti.sh ├── train_hrnet18_p3.8xlarge.sh ├── train_hrnet34_p3.8xlarge.sh ├── train_hrnet34_unet64_aug_hard.sh ├── train_hrnet34_unet64_aug_light.sh ├── train_hrnet34_unet64_aug_medium.sh ├── train_hrnet34_unet64_aug_none.sh ├── train_hrnet48_p3.8xlarge.sh ├── train_mixnet.sh ├── train_resnets.sh ├── train_seresnext50_can.sh └── train_u2net.sh ├── submit.py └── tests ├── mask.png ├── test_data.py └── test_models.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/README.md -------------------------------------------------------------------------------- /black.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/black.toml -------------------------------------------------------------------------------- /compress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/compress.py -------------------------------------------------------------------------------- /cut_test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/cut_test_dataset.py -------------------------------------------------------------------------------- /fit_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/fit_predict.py -------------------------------------------------------------------------------- /fit_predict2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/fit_predict2.py -------------------------------------------------------------------------------- /inria/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/augmentations.py -------------------------------------------------------------------------------- /inria/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/dataset.py -------------------------------------------------------------------------------- /inria/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/factory.py -------------------------------------------------------------------------------- /inria/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/losses.py -------------------------------------------------------------------------------- /inria/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/metric.py -------------------------------------------------------------------------------- /inria/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/models/__init__.py -------------------------------------------------------------------------------- /inria/models/can.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/models/can.py -------------------------------------------------------------------------------- /inria/models/deeplab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/models/deeplab.py -------------------------------------------------------------------------------- /inria/models/efficient_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/models/efficient_unet.py -------------------------------------------------------------------------------- /inria/models/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/models/fpn.py -------------------------------------------------------------------------------- /inria/models/hg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/models/hg.py -------------------------------------------------------------------------------- /inria/models/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/models/hrnet.py -------------------------------------------------------------------------------- /inria/models/u2net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/models/u2net.py -------------------------------------------------------------------------------- /inria/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/models/unet.py -------------------------------------------------------------------------------- /inria/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/optim.py -------------------------------------------------------------------------------- /inria/pseudo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/pseudo.py -------------------------------------------------------------------------------- /inria/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/scheduler.py -------------------------------------------------------------------------------- /inria/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/inria/visualization.py -------------------------------------------------------------------------------- /make_train_tiles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/make_train_tiles.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_tensorboard.cmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/run_tensorboard.cmd -------------------------------------------------------------------------------- /run_tensorboard.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/run_tensorboard.sh -------------------------------------------------------------------------------- /sample_color.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/sample_color.jpg -------------------------------------------------------------------------------- /scripts/train.cmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train.cmd -------------------------------------------------------------------------------- /scripts/train_b6.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_b6.sh -------------------------------------------------------------------------------- /scripts/train_hrnet18_4x1080Ti.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_hrnet18_4x1080Ti.sh -------------------------------------------------------------------------------- /scripts/train_hrnet18_p3.8xlarge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_hrnet18_p3.8xlarge.sh -------------------------------------------------------------------------------- /scripts/train_hrnet34_p3.8xlarge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_hrnet34_p3.8xlarge.sh -------------------------------------------------------------------------------- /scripts/train_hrnet34_unet64_aug_hard.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_hrnet34_unet64_aug_hard.sh -------------------------------------------------------------------------------- /scripts/train_hrnet34_unet64_aug_light.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_hrnet34_unet64_aug_light.sh -------------------------------------------------------------------------------- /scripts/train_hrnet34_unet64_aug_medium.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_hrnet34_unet64_aug_medium.sh -------------------------------------------------------------------------------- /scripts/train_hrnet34_unet64_aug_none.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_hrnet34_unet64_aug_none.sh -------------------------------------------------------------------------------- /scripts/train_hrnet48_p3.8xlarge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_hrnet48_p3.8xlarge.sh -------------------------------------------------------------------------------- /scripts/train_mixnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_mixnet.sh -------------------------------------------------------------------------------- /scripts/train_resnets.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_resnets.sh -------------------------------------------------------------------------------- /scripts/train_seresnext50_can.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_seresnext50_can.sh -------------------------------------------------------------------------------- /scripts/train_u2net.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/scripts/train_u2net.sh -------------------------------------------------------------------------------- /submit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/submit.py -------------------------------------------------------------------------------- /tests/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/tests/mask.png -------------------------------------------------------------------------------- /tests/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/tests/test_data.py -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BloodAxe/Catalyst-Inria-Segmentation-Example/HEAD/tests/test_models.py --------------------------------------------------------------------------------