LICENSE
README.md
setup.py
pytorch_segmentation_models_trainer/__init__.py
pytorch_segmentation_models_trainer/build_mask.py
pytorch_segmentation_models_trainer/config_utils.py
pytorch_segmentation_models_trainer/main.py
pytorch_segmentation_models_trainer/predict.py
pytorch_segmentation_models_trainer/train.py
pytorch_segmentation_models_trainer.egg-info/PKG-INFO
pytorch_segmentation_models_trainer.egg-info/SOURCES.txt
pytorch_segmentation_models_trainer.egg-info/dependency_links.txt
pytorch_segmentation_models_trainer.egg-info/entry_points.txt
pytorch_segmentation_models_trainer.egg-info/requires.txt
pytorch_segmentation_models_trainer.egg-info/top_level.txt
pytorch_segmentation_models_trainer/conf/__init__.py
pytorch_segmentation_models_trainer/custom_callbacks/__init__.py
pytorch_segmentation_models_trainer/custom_callbacks/image_callbacks.py
pytorch_segmentation_models_trainer/custom_callbacks/training_callbacks.py
pytorch_segmentation_models_trainer/custom_losses/__init__.py
pytorch_segmentation_models_trainer/custom_losses/base_loss.py
pytorch_segmentation_models_trainer/custom_losses/crossfield_losses.py
pytorch_segmentation_models_trainer/custom_losses/loss_config_definition.py
pytorch_segmentation_models_trainer/custom_metrics/__init__.py
pytorch_segmentation_models_trainer/custom_metrics/metrics.py
pytorch_segmentation_models_trainer/custom_models/__init__.py
pytorch_segmentation_models_trainer/custom_models/models.py
pytorch_segmentation_models_trainer/custom_models/unet.py
pytorch_segmentation_models_trainer/custom_models/unet_resnet.py
pytorch_segmentation_models_trainer/custom_models/utils.py
pytorch_segmentation_models_trainer/custom_models/hrnet_models/__init__.py
pytorch_segmentation_models_trainer/custom_models/hrnet_models/bn_helper.py
pytorch_segmentation_models_trainer/custom_models/hrnet_models/hrnet.py
pytorch_segmentation_models_trainer/custom_models/hrnet_models/seg_hrnet.py
pytorch_segmentation_models_trainer/custom_models/hrnet_models/seg_hrnet_ocr.py
pytorch_segmentation_models_trainer/custom_models/hrnet_models/conf/__init__.py
pytorch_segmentation_models_trainer/dataset_loader/__init__.py
pytorch_segmentation_models_trainer/dataset_loader/dataset.py
pytorch_segmentation_models_trainer/model_loader/__init__.py
pytorch_segmentation_models_trainer/model_loader/frame_field_model.py
pytorch_segmentation_models_trainer/model_loader/model.py
pytorch_segmentation_models_trainer/optimizers/__init__.py
pytorch_segmentation_models_trainer/optimizers/poly_optimizers.py
pytorch_segmentation_models_trainer/tools/__init__.py
pytorch_segmentation_models_trainer/tools/data_readers/__init__.py
pytorch_segmentation_models_trainer/tools/data_readers/raster_reader.py
pytorch_segmentation_models_trainer/tools/data_readers/vector_reader.py
pytorch_segmentation_models_trainer/tools/mask_building/__init__.py
pytorch_segmentation_models_trainer/tools/mask_building/mask_builder.py
pytorch_segmentation_models_trainer/tools/parallel_processing/__init__.py
pytorch_segmentation_models_trainer/tools/parallel_processing/process_executor.py
pytorch_segmentation_models_trainer/tools/polygonization/__init__.py
pytorch_segmentation_models_trainer/tools/polygonization/polygonize_utils.py
pytorch_segmentation_models_trainer/tools/polygonization/polygonizer_factory.py
pytorch_segmentation_models_trainer/tools/polygonization/skeletonize_tensor_tools.py
pytorch_segmentation_models_trainer/tools/polygonization/methods/__init__.py
pytorch_segmentation_models_trainer/tools/polygonization/methods/abstract_polygonizer.py
pytorch_segmentation_models_trainer/tools/polygonization/methods/active_contours.py
pytorch_segmentation_models_trainer/tools/polygonization/methods/active_skeletons.py
pytorch_segmentation_models_trainer/tools/visualization/__init__.py
pytorch_segmentation_models_trainer/tools/visualization/base_plot_tools.py
pytorch_segmentation_models_trainer/tools/visualization/crossfield_plot.py
pytorch_segmentation_models_trainer/utils/__init__.py
pytorch_segmentation_models_trainer/utils/complex_utils.py
pytorch_segmentation_models_trainer/utils/frame_field_utils.py
pytorch_segmentation_models_trainer/utils/math_utils.py
pytorch_segmentation_models_trainer/utils/model_utils.py
pytorch_segmentation_models_trainer/utils/os_utils.py
pytorch_segmentation_models_trainer/utils/polygon_utils.py
pytorch_segmentation_models_trainer/utils/tensor_utils.py
tests/test_configs/__init__.py