Can onnx model generate masks for an entire image？

ChaoningZhang / MobileSAM

This is the official code for MobileSAM project that makes SAM lightweight for mobile applications and beyond!

Apache License 2.0

4.86k stars 505 forks source link

Can onnx model generate masks for an entire image？ #83

Open tianjiahao opened 1 year ago

tianjiahao commented 1 year ago

onnx_model_example.ipynb this support point_coords parameter ，I not find onnx model generate masks for an entire image like this： from mobile_sam import SamAutomaticMaskGenerator

mask_generator = SamAutomaticMaskGenerator(mobile_sam) masks = mask_generator.generate()