Skip to content

I did semantic segmentation on a custom dataset by the pretrained resNet50 model in PyTorch. I also did a comparative analysis of model(resNet50) configuration with and without skip connections

License

Notifications You must be signed in to change notification settings

sghawana/Image-segmentation-by-pretrained-ResNet50

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Semantic Segmentation

This project demonstrates the use of a pretrained ResNet50 model for semantic image segmentation.
The goal is to identify and separate different objects or regions within an image by applying a segmentation model.


Dataset

The dataset consis of real images and their respective ground truth masks.
Each image can be segmented into at most 9 different regions

Example image

mask


Architecture

It utilizes a ResNet50 model pre-trained on ImageNet and fine-tuned for semantic segmentation using a U-net.

The model is tested on a dataset of images, and segmentation masks are predicted to classify different regions of the images.

segm


Results

Please refer to report.pdf for detailed results and analysis

Some sample model predictions

res1
res2


About

I did semantic segmentation on a custom dataset by the pretrained resNet50 model in PyTorch. I also did a comparative analysis of model(resNet50) configuration with and without skip connections

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published