How do I create a Image Segmentation Dataset

As always, become one with the data. Make sure the data is prepared in the right way.

You can do the following things:

  • print examples of your dataset
  • create a PyTorch DataLoader and check whether the data is created appropriately
1 Like