want to get image mask, for use in bundlesdf project, just run demo line,

[wang-shuaikang]

No description provided.

[m43]

Hej, if you're looking to extract the mask logits from our demo, you can find the relevant line of code below. Is this what you were searching for?

sam-pt/demo/demo.py

Line 134 in 228d7cd

logits_list = outputs['logits']

[2min77]

@m43 Hello Sir, after normal operation, should there be a mask in the form of pictures in the output folder? There is no mask picture in my output folder, thank you for your answer

[m43]

For the demo unfortunately no, just the visualizations are created, but you can access the logits here. When running SAM-PT on VOS as documented here, the masks will automatically be saved to disk in the form of pictures as they are later needed for computing the VOS performance. You can see how in that case we process the logits here and save them to disk here.