This example shows how to train a semantic segmentation deep learning network using your own dataset. In this example, I will demonstrate how to label the pixel in the image by using MATAB image labeler app.After completing the labelling, I will export the labelling to workspace as 'gTruth'.
Later, I modify example below to accept gTruth as dataset.
After my modification, you do not need to modify anything, it would be workable if you run them directly. However, if the accuracy of network is not satisfied, you may tune the network with different hyperparameter setting and network selection.
1) Label your image at pixel level by MATLAB image labeler app
2) Concept and workflow of semantic segmentation using deep learning
3) Create two datastore (Image datastore and pixel Label datastore)
4) Modify Vgg16 or Vgg19 to SegNet
5) Classify the image by trained SegNet
Product Focus :
Deep Learning Toolbox
Written at 26 February 2019
Kevin Chng (2021). Getting Started with Semantic Segmentation using DL (https://www.mathworks.com/matlabcentral/fileexchange/70400-getting-started-with-semantic-segmentation-using-dl), MATLAB Central File Exchange. Retrieved .
MATLAB Release Compatibility
Platform CompatibilityWindows macOS Linux
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!Start Hunting!