how to train on my own dataset ?

usuyama / pytorch-unet

Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation

MIT License

827 stars 233 forks source link

Open manvirvirk opened 4 years ago

manvirvirk commented 4 years ago

can anyone help?

usuyama commented 4 years ago

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

Some related discussons: