Skip to content
This repository has been archived by the owner on Jun 26, 2021. It is now read-only.

First Beta Release

Pre-release
Pre-release
Compare
Choose a tag to compare
@justusschock justusschock released this 15 Jan 14:03
· 1257 commits to master since this release
6ab0a1d

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