How to use masks prompt

[wu2233]

Hello, I hope to use a binary image (with a shape of 320, 480) as masks prompt input into the prompt_encoder. I have transformed it into (1, 1, 320, 480), but the dense_embedding output by the prompt_encoder cannot be input into the mask_decoder. I would like to know what processing I need to do to this binary image so that it can become a masks prompt.