BingDwenDwen icon indicating copy to clipboard operation
BingDwenDwen copied to clipboard

hello, maybe something wrong in MODNet

Open ezio1320 opened this issue 1 year ago • 1 comments

你好,不好意思在这个项目里打扰你。 感谢你的modnet的训练代码,非常有帮助,但是我在训练的时候发现detail loss非常的大。我检查了相关代码,应该是trimap生成的部分有一点小小的问题。在src/datasets/ppm100.py中的line 45 到line 47,将line45 行matte = matte_transform(matte) 置后才能得到正确的trimap。 再次感谢你分享的代码,祝好!

ezio1320 avatar Mar 20 '23 11:03 ezio1320

好的 感谢指正

Fight-hawk avatar Mar 22 '23 08:03 Fight-hawk