Projects

nnDetection: A Self-configuring Method for Medical Object Detection

Simultaneous localisation and categorization of objects in medical images, also referred to as medical object detection, is of high clinical relevance because diagnostic decisions depend on rating of objects rather than e.g. pixels. For this task, the cumbersome and iterative process of method configuration constitutes a major research bottleneck. Recently, nnU-Net has tackled this challenge for the task of image segmentation with great success. Following nnU-Net’s agenda, in this work we systematize and automate the configuration process for medical object detection. The resulting self-configuring method, nnDetection, adapts itself without any manual intervention to arbitrary medical detection problems while achieving results en par with or superior to the state-of-the-art. We demonstrate the effectiveness of nnDetection on two public benchmarks, ADAM and LUNA16, and propose 10 further public data sets for a comprehensive evaluation of medical object detection methods.

Rising: High-Performance Data Loading and Augmentation for 2D & 3D Images

Rising is a high-performance data loading and augmentation library for 2D and 3D data completely written in PyTorch. Our goal is to provide a seamless integration into the PyTorch Ecosystem without sacrificing usability or features. All transformations are directly implemented in PyTorch which allows gradient propagation and direct execution on GPUs. The provided dataloading module allows for various types of transformations (CPU vs GPU, per sample vs batched) to maximise the resuability of all implemented components.

Delira - A Backend Agnostic High Level Deep Learning Library

Lightweight framework for fast prototyping and training of deep neural networks with PyTorch and TensorFlow. delira is designed to work as a backend agnostic high level deep learning library which allwos the user to compare various models written in different backends without rewriting them.