On the calculation of FLOPs for SAM/MobileSAM

[BeliverK]

SAM model requires multiple inputs, image, point_prompt or boxes_prompt, how do you pass these parameters to the calculate_flops?For example, in the following code, how to set parameters？

from mobile_sam import MobileSAM, SamPredictor
import torch

model = MobileSAM.from_pretrained("nielsr/mobilesam")

# perform inference
device = "cuda" if torch.cuda.is_available() else "cpu"
model.to(device=device)

predictor = SamPredictor(model)
predictor.set_image(<your_image>)
masks, _, _ = predictor.predict(<input_prompts>)