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

question about mask dimension #14

Open
spacewalk01 opened this issue Dec 14, 2023 · 3 comments
Open

question about mask dimension #14

spacewalk01 opened this issue Dec 14, 2023 · 3 comments

Comments

@spacewalk01
Copy link

Thank you for your work! The output mask shape is 4x256x256 where 4 is (I guess) the number of labels and 256x256 is mask's height x width dimensions. I wonder how to get 1x256x256 mask from the output?

@spacewalk01
Copy link
Author

spacewalk01 commented Dec 14, 2023

you are performing (mask[0, 0] > 0).detach().cpu().numpy() on 1 x 4 x 256 x 256 matrix which means that first only 256 x 256 returns ignoring the rest, is it background mask?

@spacewalk01
Copy link
Author

I am reimplementing it but in my implementation, it gets me incorrect results on the mask detection task given a bounding box.

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

1 participant