My architecture implementation from paper U-Net: Convolutional Networks for Biomedical Image Segmentation

using package manager uv:

uv venv && uv sync

using python venv:

python3 -m venv .venv && source .venv/bin/activate
python3 -m pip install .

You can train the model using script:

python3 train.py [here arguments for script defined in script]

• Add option to train with Dice Loss