U-Net is an architecture for semantic segmentation. It consists of a contracting path and an expansive path. The contracting path follows the typical architecture of a convolutional network. It consists of the repeated application of two 3x3 convolutions (unpadded convolutions), each followed by a rectified linear unit (ReLU) and a 2x2 max pooling operation with stride 2 for downsampling. At each downsampling step we double the number of feature channels. Every step in the expansive path consists of an upsampling of the feature map followed by a 2x2 convolution (“up-convolution”) that halves the number of feature channels, a concatenation with the correspondingly cropped feature map from the contracting path, and two 3x3 convolutions, each followed by a ReLU. The cropping is necessary due to the loss of border pixels in every convolution. At the final layer a 1x1 convolution is used to map each 64-component feature vector to the desired number of classes. In total the network has 23 convolutional layers.
Source: U-Net: Convolutional Networks for Biomedical Image SegmentationPaper | Code | Results | Date | Stars |
---|
Task | Papers | Share |
---|---|---|
Semantic Segmentation | 118 | 13.56% |
Image Segmentation | 83 | 9.54% |
Medical Image Segmentation | 54 | 6.21% |
Denoising | 52 | 5.98% |
Image Generation | 37 | 4.25% |
Tumor Segmentation | 25 | 2.87% |
Brain Tumor Segmentation | 13 | 1.49% |
Lesion Segmentation | 13 | 1.49% |
Computed Tomography (CT) | 12 | 1.38% |
Component | Type |
|
---|---|---|
Concatenated Skip Connection
|
Skip Connections | |
Convolution
|
Convolutions | |
Max Pooling
|
Pooling Operations | |
ReLU
|
Activation Functions |