search

THU-MIG / RepViT

RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
https://arxiv.org/abs/2307.09283
Apache License 2.0
681 stars 55 forks source link

make onnx export like samexporter #28

Closed ZHEQIUSHUI closed 7 months ago

jameslahm commented 7 months ago

Thanks for your PR. This would likely break the CoreML export due to the interpolation with dynamic size. I'd appreciate it if you could split the usage of SamOnnxModel from the current CoreML export, like adding a SamCoreMLModel separately for the CoreML export using the current sam/repvit_sam/utils/onnx.py and keeping the SamOnnxModel the same as your PR.

ZHEQIUSHUI commented 7 months ago

does coreml model need "orig_im_size"?

jameslahm commented 7 months ago

Thanks. Coreml models don't need "orig_im_size".