search

rishizek / tensorflow-deeplab-v3-plus

DeepLabv3+ built in TensorFlow
MIT License
834 stars 307 forks source link

How to adjust label weights? #87

Closed 476117957 closed 4 years ago

476117957 commented 4 years ago

For example the proportion of the pixels of background, class1, class2 is 10:5:1, how to adjust the weight of each label in loss? Thanks a lot!

476117957 commented 4 years ago

I think I found it in deeplab_model.py line 220.