Skip to content

vishalsingha/Image-Segmentation-U-net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

51 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image-Segmentation-U-net

Dataset

  • Source : https://www.robots.ox.ac.uk/~vgg/data/pets/
  • Discription:
    Dataset Consist of colorder images and annonation correcponding to each images. The annonations are 2D png images containing 1's, 2's, 3's where
    1's : foreground
    2's : boundary
    3's : background

Data preparation:

All the images are resized to (256, 256, 3) dimention by using tf.data api to make datagenerator. And the annonation were also converted from (256, 256) to (256, 256, 3) images by using pixel level one hot encoding.

Model used:

U-net model having (None, 256, 256, 3) as input dim and (None 256, 256, 3) as output dimention.

Results :

  • Default train and test size (Segmentation_Unet.ipynb) :
    • train size: 2944
    • validation size : 736
    • test size : 3369

  • Custom train and test size (segmentation_Unet_final.ipynb) :
    • train size : 5621
    • validatio size : 1103
    • test size : 625

Varification on test data (real vs predicted)

  • Default train and test size

    • Some good results:

      image image image image image image

    • some bad results

      image image image image image image

  • Custom train and test size

    image image image image image image image image

Conclusion:

The accuracy of custom train test split increase the accuracy of prediction alot due to the increase of training data. If we increase the size of training set the model is expected to give better performance.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published