├── .github └── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md ├── .gitignore ├── CITATION.cff ├── CODE_OF_CONDUCT.md ├── LICENSE ├── README.md ├── batch_train_models.py ├── batch_train_models_no_tkinter.py ├── commands_to_run_scripts_docker.md ├── compose.yml ├── dockerfile ├── entrypoint.sh ├── install ├── gym.yml └── gym_test_gpus.dockerfile ├── make_dataset.py ├── make_dataset_no_tkinter.py ├── pixi.lock ├── pyproject.toml ├── seg_images_in_folder.py ├── src └── segmentation_gym │ └── __init__.py ├── train_model.py ├── train_model_script_no_tkinter.py └── utils ├── batch_pred2map.py ├── filter_bad_tensors.py ├── filter_labels.py ├── gen_fullmodel_from_h5.py ├── gen_saved_model.py ├── make_class_balanced_subset.py ├── make_overlays.py ├── pred2map.py ├── preprocess_data.py ├── print_pred_labels.py ├── sort_labels_class_balance.py ├── test_gpus.py └── vanilla_unet_config └── vanilla_unet_ronneburger2015.json /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/CITATION.cff -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/README.md -------------------------------------------------------------------------------- /batch_train_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/batch_train_models.py -------------------------------------------------------------------------------- /batch_train_models_no_tkinter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/batch_train_models_no_tkinter.py -------------------------------------------------------------------------------- /commands_to_run_scripts_docker.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/commands_to_run_scripts_docker.md -------------------------------------------------------------------------------- /compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/compose.yml -------------------------------------------------------------------------------- /dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/dockerfile -------------------------------------------------------------------------------- /entrypoint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/entrypoint.sh -------------------------------------------------------------------------------- /install/gym.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/install/gym.yml -------------------------------------------------------------------------------- /install/gym_test_gpus.dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/install/gym_test_gpus.dockerfile -------------------------------------------------------------------------------- /make_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/make_dataset.py -------------------------------------------------------------------------------- /make_dataset_no_tkinter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/make_dataset_no_tkinter.py -------------------------------------------------------------------------------- /pixi.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/pixi.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/pyproject.toml -------------------------------------------------------------------------------- /seg_images_in_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/seg_images_in_folder.py -------------------------------------------------------------------------------- /src/segmentation_gym/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/train_model.py -------------------------------------------------------------------------------- /train_model_script_no_tkinter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/train_model_script_no_tkinter.py -------------------------------------------------------------------------------- /utils/batch_pred2map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/batch_pred2map.py -------------------------------------------------------------------------------- /utils/filter_bad_tensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/filter_bad_tensors.py -------------------------------------------------------------------------------- /utils/filter_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/filter_labels.py -------------------------------------------------------------------------------- /utils/gen_fullmodel_from_h5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/gen_fullmodel_from_h5.py -------------------------------------------------------------------------------- /utils/gen_saved_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/gen_saved_model.py -------------------------------------------------------------------------------- /utils/make_class_balanced_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/make_class_balanced_subset.py -------------------------------------------------------------------------------- /utils/make_overlays.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/make_overlays.py -------------------------------------------------------------------------------- /utils/pred2map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/pred2map.py -------------------------------------------------------------------------------- /utils/preprocess_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/preprocess_data.py -------------------------------------------------------------------------------- /utils/print_pred_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/print_pred_labels.py -------------------------------------------------------------------------------- /utils/sort_labels_class_balance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/sort_labels_class_balance.py -------------------------------------------------------------------------------- /utils/test_gpus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/test_gpus.py -------------------------------------------------------------------------------- /utils/vanilla_unet_config/vanilla_unet_ronneburger2015.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Doodleverse/segmentation_gym/HEAD/utils/vanilla_unet_config/vanilla_unet_ronneburger2015.json --------------------------------------------------------------------------------