Open lucy3589 opened 5 years ago
Hi Lucy,
I think you'd better ask to the original code author (https://github.com/jakeret/tf_unet). I haven't used the cross-entropy loss for my problems as they seem to vulnerable to imbalanced ones.
ok,thanks very much!
The separation border is computed using morphological operations. The weight map is then computed as w(x) = wc(x) + w0 · exp −(d1(x) + d2(x))2 2σ2 ! (2) where wc : Ω ! R is the weight map to balance the class frequencies, d1 : Ω ! R denotes the distance to the border of the nearest cell and d2 : Ω ! R the distance to the border of the second nearest cell. In our experiments we set w0 = 10 and σ ≈ 5 pixels.