UNet3plus_pth icon indicating copy to clipboard operation
UNet3plus_pth copied to clipboard

hi train with my own data

Open zhangyunming opened this issue 4 years ago • 5 comments

i train the v3 with my own data that with lables 0:backgrounbd and 1:portrait, just two classes ,so the n_classes=1 , but you masks labels are 1-255 , and my train loss is not change and the test result are all black 0 . so how can i do .thanks

zhangyunming avatar Jul 13 '20 11:07 zhangyunming

the bug is ok.its similar with pytorch-unet

zhangyunming avatar Jul 14 '20 06:07 zhangyunming

hey, you maybe change into UNet2plus to try. If the error is same, there is wrong in your data or using loss function. Else there is wrong in UNet3plus model structure.

avBuffer avatar Jul 15 '20 02:07 avBuffer

Hey,thanks! I tried it and found the training loss was changed after about 18 steps. In fact, from step-0 to step-17, the training loss had a bit a bit changed. You maybe train a long steps.

avBuffer avatar Jul 15 '20 06:07 avBuffer

I train myself on data, DICE is low and constant, how do I solve it

jialeqaq avatar Jul 07 '23 10:07 jialeqaq

i train the v3 with my own data that with lables 0:backgrounbd and 1:portrait, just two classes ,so the n_classes=1 , but you masks labels are 1-255 , and my train loss is not change and the test result are all black 0 . so how can i do .thanks

hello,why my training data DICE is very low and the more I train, the worse it gets

jialeqaq avatar Jul 08 '23 10:07 jialeqaq