- This repository is used to provide a code base for PyTorch-based training framework. One can easily develop their own model on top of it.
- Some dirty codes are still there without debugging.
- UNet-based denoising/segmentation examples are provided based on two open-access datasets
- Denoising: AAPM LowDose CT Challenge dataset
- Segmentation: StructSeg segmentation dataset.
- customize your own dataset within lib/dataset
- customize your own specific training workflow within lib/model
- customize your own architecture within lib/model/module