yucornetto/MGMatting

Questions about loss weight?

mingqizhang opened this issue · 0 comments

Hi, I have some questions about the code?
The regression loss and lap loss were only focus on the weight( 1/255<uncertain area <1-1/255) part of the prediction, which means the model not learn how to predict background(<1/255) and foreground(>1-1/255). I delet the "weight_os8[...] = 1" in trainer.py to ensure that weight_os8, weight_os4 and weight_os1 only have uncertain area value. In my opinion, the prediction in foreground and background should have the random value. But the model can also predict foreground and background.