/U-Net_on_OxfordIIITPet

This is small project, for study U-Net architecture

Primary LanguagePythonApache License 2.0Apache-2.0

U-Net_on_OxfordIIITPet

This is small project, for study U-Net-style architecture

Dataset

For this project I used Oxford pets dataset. This dataset has 7930 images of dogs and cats, with mask for segmentation training.

Model

Model is simply U-Net-style network, with encoder based on convolution-relu-batchnorm block, and max pooling, and decoder with deconvolution-convolution and concatenation step.

Result

I trained this network on Kaggle kernel with Tesla P100 GPU. After 15 epochs, results are pretty neat.

Ground truth

Abyssian

Segmentation

Abyssian

Target mask

Abyssian

Ground truth

Pug

Segmentation

Pug

Target mask

Pug

Ground truth

Bulldog

Segmentation

Bulldog

Target mask

Bulldog