/dynamic-network-architectures

Primary LanguagePythonApache License 2.0Apache-2.0

Dynamic Network Architectures

This repository contains several ResNet, U-Net and VGG architectures in pytorch that can be dynamically adapted to a varying number of image dimensions (1D, 2D or 3D) and the number of input channels.

Available models

ResNet

We implement the standard ResNetD 18, 34, 50 and 152. For ResNets 50 and 152 also bottleneck implementations are available. Moreover, adapted versions that are better suited for smaller image sizes such as CIFAR can be used.

All models additionally include regularization techniques like Stochastic Depth, Squeeze & Excitation and Final Layer Dropout.

VGG

In contrast to the original VGG implementation we exclude the final fully-connected layers in the end and replace it by additional convolutional layers and only one fully-connected layer in the end. Adapted versions that are better suited for smaller image sizes such as CIFAR can be used.

U-Net

For the U-Net a plain convolutional encoder as well as a residual encoder are available.

Acknowledgements

    

This Repository is developed and maintained by the Applied Computer Vision Lab (ACVL) of Helmholtz Imaging.