This repository has been archived by the owner on Jun 26, 2021. It is now read-only.
First Beta Release
Pre-release
Pre-release
justusschock
released this
15 Jan 14:03
·
1257 commits
to master
since this release
Based on PyTorch, batchgenerators and trixi it provides a framework for
- Dataset loading
- Dataset sampling
- Augmentation (multi-threaded) including 3D images with any number of channels
- A generic trainer class that implements the training process
- Already implemented models used in medical image processing and exemplaric implementations of most used models in general (like Resnet)
- Web-based monitoring using Visdom
- Model save and load functions