Skip to content

Latest commit

 

History

History
5 lines (3 loc) · 580 Bytes

File metadata and controls

5 lines (3 loc) · 580 Bytes

Oxford_IIIT_Pets_dataset_image_segmentation

The model being used here is a modified U-Net. A U-Net consists of an encoder (downsampler) and decoder (upsampler). In-order to learn robust features, and reduce the number of trainable parameters, a pretrained model can be used as the encoder. Thus, the encoder for this task will be a pretrained MobileNetV2 model, whose intermediate outputs will be used, and the decoder will be the upsample block already implemented in TensorFlow Examples in the Pix2pix tutorial.

After training, predictions will be made on the test dataset.