Open spacewalk01 opened 8 months ago
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?
I am reimplementing it but in my implementation, it gets me incorrect results on the mask detection task given a bounding box.
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?