NVlabs / SegFormer

Official PyTorch implementation of SegFormer
https://arxiv.org/abs/2105.15203
Other
2.5k stars 349 forks source link

How to convert the model to tensorrt or onnx? #114

Open yuchenlichuck opened 1 year ago

yuchenlichuck commented 1 year ago

For robot implementation, we need onnx or openvino version of segformer, but currently, I found that segformer is not to to be converted to those versions, Does anyone can help us to find the reason or share your successful examples, thank you!

LeonArcher commented 1 year ago

Was able to do this with a SegFormer hosted on Huggingface Transformers (function transformers.onnx.export).

xueyingliu commented 1 year ago

@yuchenlichuck I have finished to export onnx model, you can refer to this.