pytorch-unet icon indicating copy to clipboard operation
pytorch-unet copied to clipboard

Published 20 hours ago • verified publisher icon usuyama

how to train on my own dataset ?

Open manvirvirk opened this issue 5 years ago • 1 comments

can anyone help?

manvirvirk avatar Mar 19 '20 17:03 manvirvirk

You can create a new Dataset class similar to the SimDataset class.

Some related discussons:

  • https://discuss.pytorch.org/t/how-make-customised-dataset-for-semantic-segmentation/30881/16

usuyama avatar May 19 '20 06:05 usuyama