I think the boundary mask is calculated wrong

[ezio1320]

hello,zhanghan: Modnet is a exciting work in matting research, and your code is helpful a lot. Thank you ! In the process of using your code, I found that there are some problems with the calculation of the bound mask. In trainer.py , Line 146. the mask calculated as follow: boundaries = (trimap < 0.5) + (trimap > 0.5)

I think the "boundary" means the unknown area of trimap from the paper and literally. So, boundaries = (trimap==0.5) Which is right? Does anyone else have same question?