Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implementation of PRM module #11

Open
wds1998 opened this issue Jun 21, 2021 · 2 comments
Open

Implementation of PRM module #11

wds1998 opened this issue Jun 21, 2021 · 2 comments

Comments

@wds1998
Copy link

wds1998 commented Jun 21, 2021

Hello, your work MGMatting is very excellent. I read your paper and code, I found that the PRM module code in the paper is implemented using two-layer convolution. Do you have any code for PRM in the paper? Does PRM bring a great gain effect when you make experiments, especially for the details such as the hairline of the portrait matting? Thank you very much for answering this question.

@yucornetto
Copy link
Owner

Hi, please refer to the decoder file in the code-base, where additional convs of PRM are defined. But the usage of PRM also include some code parts in utils.py and main.py, in terms of how to use later-stage prediction to refine the previous stage's one, and also related loss computation.

@wds1998
Copy link
Author

wds1998 commented Jul 8, 2021

Hi, please refer to the decoder file in the code-base, where additional convs of PRM are defined. But the usage of PRM also include some code parts in utils.py and main.py, in terms of how to use later-stage prediction to refine the previous stage's one, and also related loss computation.

Thank you very much.In addition, I would like to ask if you encountered a situation where the background pixel was incorrectly predicted as the foreground pixel during the test (the value of the background pixel is predicted to be a low confidence value), do you know how to solve it or how to strengthen the guidance of mask for the final alpha result.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants