TransWikia.com

How can I use my own dataset for Image segmentation using Tensorflow

Data Science Asked by thegendolz on December 24, 2020

I have a huge problem using my own created dataset for image segmentation using Tensorflow. The dataset that I’ve build contain images like the one shown below:

Dataset example

The problem that I have is: How do I use my own dataset specifically for image segmentation? I’ve looked at the documentation on how to create datasets but all the examples either only use object detection with a single class or classify the entire image. I want to assign every pixel to a class (image segmentation) and use it to train my model.

I’ve also found examples of image segmentation but they all use existing datasets such as Cityscapes, ADE20k etc. How can I use my own images and data and transform it to a tensorflow dataset so that I can use it for training?

One Answer

Just having segmented images is probably not enough. The training data for segmentation needs to be in a specific format. Have a look at the coco dataset for image segmentation. Sometimes we need to convert the dataset into that format. I'd suggest reading up a bit on how to train a mask rcnn model on your own dataset. There are a number of articles available online.

Answered by tehem on December 24, 2020

Add your own answers!

Ask a Question

Get help from others!

© 2024 TransWikia.com. All rights reserved. Sites we Love: PCI Database, UKBizDB, Menu Kuliner, Sharing RPP