search

hustvl / ViTMatte

[Information Fusion (Vol.103, Mar. '24)] Boosting Image Matting with Pretrained Plain Vision Transformers
MIT License
339 stars 33 forks source link

训练时的trimap怎么来的 #25

Closed WHHWW closed 10 months ago

WHHWW commented 10 months ago

作者你好，很抱歉打扰到您，我一直想知道训练时的trimap是怎么获得的，我看的您的模型里需要输入原图和trimap，但是我并没有找到trimap的路径，composition-1K的训练集里也没有trimap，请问您的项目里训练时的trimap是哪里来的呢，希望能够得到您的解答，万分感谢！

JingfengYao commented 10 months ago

是根据alpha gt在线生成的伪trimap。你可以参见这里的代码 https://github.com/hustvl/ViTMatte/blob/main/data/dim_dataset.py#L498

WHHWW commented 10 months ago

是根据alpha gt在线生成的伪trimap。你可以参见这里的代码 https://github.com/hustvl/ViTMatte/blob/main/data/dim_dataset.py#L498 好的好的，感谢您的回复！