- /configs/default.json
- /docs/variants.md
- /scripts/render.py
- /scripts/train.py
- /src/zhu_unet/__init__.py
- /src/zhu_unet/attention.py
- /src/zhu_unet/augment.py
- /src/zhu_unet/datasets.py
- /src/zhu_unet/features.py
- /src/zhu_unet/losses.py
- /src/zhu_unet/models.py
- /src/zhu_unet/renderer.py
- /src/zhu_unet/utils.py
- /tests/test_smoke.py
- /LICENSE
- /pyproject.toml
- /README.md
- /requirements.txt
- /.gitignore
- /CITATION.cff