Segmentation Data Module
========================

The `data_module` module provides a PyTorch Lightning `LightningDataModule` for semantic segmentation tasks. 

.. automodule:: ucs.data.data_module
   :members:
   :undoc-members:
   :show-inheritance:
   :exclude-members: prepare_data, setup, train_dataloader, val_dataloader, test_dataloader