This will show how U-net works to segment the image
U-Net is the model use to create image segmentation i.e Labelling each pixel of image
- Encoding part : At first image is taken by conv_block() which downsamples the image by performing normal convolution and return next_block and skip_connections, where next_block is used by upcoming convolution that also downsamples and skip_connections is used by corresponding decoding block.
- Decoding part : It takes previous layer as first parameter and skip connection as second paramter and performs Transpose convolution to upscale the image and at last convolution the number of filters of convolution is equal to number of classses to be labeled.
- U-Net Model : It uses this encoding and decoding to create the segmenetation of image .
- Decoding part : It takes previous layer as first parameter and skip connection as second paramter and performs Transpose convolution to upscale the image and at last convolution the number of filters of convolution is equal to number of classses to be labeled.