TUI-NICR / ESANet

ESANet: Efficient RGB-D Semantic Segmentation for Indoor Scene Analysis
Other
231 stars 49 forks source link

The factors of 4 loss #14

Closed andyzj2621 closed 3 years ago

andyzj2621 commented 3 years ago

I'm studying your network,it is very good and interesting.I spent much time in looking for 4 factors of 4 training loss. In other words,i can not find a,b,c,d, in All-training-loss =a loss_from_the_last_output +b loss_from_first_decodermodule +c loss_from_second_decodermodule +d loss_from_third_decodermodule .what are them and how can i change them.I'd appreciate it if you could give me a cue.Thank you very much.

mona0809 commented 3 years ago

The losses are computed here: https://github.com/TUI-NICR/ESANet/blob/main/src/utils.py#L34 The function returns a list with the 4 losses. These 4 losses are simply summed up here: https://github.com/TUI-NICR/ESANet/blob/main/train.py#L306. That means all 4 factors equal 1. Instead of summing them up, you could define weighting factors yourself.

The order of the values in the list is the following:

Let us know if you have any further questions.

andyzj2621 commented 3 years ago

Thank for you help ,you are so kind.:smile: @mona0809