Information about custom dataset

LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.

2.68k stars 595 forks source link

Closed HamzahNizami closed 5 years ago

HamzahNizami commented 5 years ago

Hi there I wanted to use your model to train my dataset and I was wondering what format I must label my dataset.

Thanks in advance

LeeJunHyun commented 5 years ago

Hi @HamzahNizami , I used .png format which is provided by ISIC challenge, but you can use any kind of format supported by pytorch.

LeeJunHyun commented 5 years ago

If you don't have further issues, I will close the issue. I hope my answer was helpful :)