- I have implemented two FCNs (FCN-32 and FCN-16) to semantically segment images using the VOCSegmentation2012 dataset.
- Semantic segmentation is a bit different from classification, where we classify each pixel as a particular class.
- I've used Transfer Learning by transfering initial weigths of VGG-16 network and replacing classification layer with convolution layers.
- mean IOU and DICE score are the evaluation metrics.
-
Notifications
You must be signed in to change notification settings - Fork 0
nikhitmago/semantic-segmentation-transfer-learning
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Semantically segmenting images using Fully Convolutional Networks based on VGG-16 network in PyTorch
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published