how to use ONNX for multi mask?

[kira-heisenberg]

How to get all the masks from the image using onnx, the notebook implementation is for a single mask. Help is appreciated

[JKelle]

The SamOnnxModel constructor takes a boolean called return_single_mask. The notebook uses return_single_mask=True but it looks like you might be able to just pass return_single_mask=False to get a multi-mask oxxn model. I haven't tried it, but I'm guessing that's enough to get the multi-mask onnx model.

[alidexiang98]

following

[DiegoMoralesBravo]

following