Skip to content

v-octal/street_image_segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Street Image Segmentation

Training a model using UNET and ResNet34 to do image segmentation on street images. Image Segmentation is the process of partitioning a digital image into multiple segments (sets of pixels, also known as image objects).

You can find more about the dataset here.

Notebook

If you are having trouble opening the notebook with github click here to view it with NBViewer.

Results

Input:

image

Output:

image

Color codes of categories

image

We can see that the predictions in this case were pretty accurate. It predicted the buildings, the trees, the vegetation, Bicyclist, Car(Auto), Road, Sign Symbol, Pole, fence. I don't think I have any complains with it.

About

Training a model using UNET and ResNet34 to do image segmentation on street images

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published