vietanhdev / samexporter

Export Segment Anything Models to ONNX
https://pypi.org/project/samexporter/
MIT License
239 stars 29 forks source link

use single point prompt mask decoder slow than pytorch, how to solve this problem? #5

Open stihuangyuan opened 1 year ago

AlessandroMondin00 commented 1 year ago

I have runned the same experiment too: with a Macbook Air M2 it takes the following times to extract image embedding with ViT-L (sam_vit_h_4b8939):

  1. ~36 seconds with pt weights
  2. ~42 seconds with onnx encoder extracted with samexporter.export_encoder
  3. ~ 39 seconds with onnx encoder quantized extracted with samexporter.export_encoder