efficientvit icon indicating copy to clipboard operation
efficientvit copied to clipboard
Published 4 months ago • verified publisher icon mit-han-lab

Export Onnx Segmentation Model

Open chuangyuan089 opened this issue 2 years ago • 2 comments

Hi, thanks for the great work! Is there an option to switch the head when exporting the onnx segmentation model? For ade20k the output shape is currently 1 x 150 x 64 x 64 but I am wondering if I can adapt it to image size (i.e. 512 x 512)

chuangyuan089 avatar Sep 19 '23 14:09 chuangyuan089

Hi, thanks for the great work! I have another question about this part, how can I export the efficientvit-sam decoder part like the vanilla sam?

Pytrick520 avatar Sep 20 '23 08:09 Pytrick520

Hi, thanks for the great work! I have another question about this part, how can I export the efficientvit-sam decoder part like the vanilla sam?

Hi, @Pytrick520 , to export the EfficientViT-SAM decoder part in a similar way to the vanilla SAM, you can refer to the method I just provided (#27). It offers a simple approach to support exporting the encoder and decoder as ONNX formats. Feel free to use that method, and if you have any further questions or need assistance, please don't hesitate to ask!"

CVHub520 avatar Sep 22 '23 01:09 CVHub520