341.341. Semantic Segmentation

Semantic Segmentation

**Problem Statement:** Implement a semantic segmentation model using a U-Net architecture to segment objects in an image. The model should take an input image and output a segmentation mask.

**Example:** If we have an input image with size 256x256, we want to output a segmentation mask with the same size.

**Constraints:** The model should be able to segment objects in the input image, and the output segmentation mask should have the same size as the input image.