Clarification Needed on Parameter Shapes for `GAP/LAP` Inference

[zhang-haojie]

Hello,

I'm currently working with the GAP or LAP and would appreciate some guidance on the shape requirements for the inference interface parameters. For example mask_pred, image, img_mst_tree.

Could you please specify the dimensions for these parameters? An example of a runnable inference script would be incredibly helpful.

Thank you for your time and assistance.