MGMatting icon indicating copy to clipboard operation
MGMatting copied to clipboard

Questions about loss weight?

Open mingqizhang opened this issue 3 years ago • 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.

mingqizhang avatar Jan 07 '22 07:01 mingqizhang