Closed DerThorsten closed 5 years ago
* The start block can be used to increase the features from `in_channels` to `initial_features` before the actual unet, the end block is used to convert the output features of the unet to the desired number of channels (@constantinpape what do you think about theses changes)
I like the idea. Didn't have much time to look at the actual changes, will take a closer look on Monday.
This PR adds the following:
Convolutions:
ConvELU
wheredim
is a keyword of the constructor, only 1D, 2D, and 3D works)UNetBase:
UNetBase
:UNet
. The start block can be used to increase the features fromin_channels
toinitial_features
before the actual unet, the end block is used to convert the output features of the unet to the desired number of channels (@constantinpape what do you think about theses changes)This is still a work in progress, the ResUnet still needs the refactoring, also the examples are not yet up-to date