search

MrGiovanni / UNetPlusPlus

[IEEE TMI] Official Implementation for UNet++
Other
2.26k stars 538 forks source link

Train own data #9

Closed Lovesuu closed 5 years ago

Lovesuu commented 5 years ago

Thanks for yours source, but can you describe more clearly how can I train own data? I'm a student and want to try with my data includes images and the corresponding mask about the object.

mahfuzmohammad commented 5 years ago

Hi @Lovesuu, An example code is already provided in the README.md (https://github.com/MrGiovanni/UNetPlusPlus#code-examples-for-your-own-data).

There you will find variables x and y. You need to load your images in x and masks into y. The value range of these variables needs to be normalized between [0, 1]. That's it!

Hope it helps!