search

raoyongming / DenseCLIP

[CVPR 2022] DenseCLIP: Language-Guided Dense Prediction with Context-Aware Prompting
520 stars 40 forks source link

An error while saving the model in ONNX format #32

Closed Virgilzzz closed 1 year ago

Virgilzzz commented 2 years ago

Have you ever tried saving the model as ONNX for visualization? Why does the trained model report an error missing img_meta when saved as ONNX?

raoyongming commented 2 years ago

Hi, thanks for your interest in our work. We don't have much experience in using ONNX. Our code is based on mmseg and mmdet. I think maybe you can check how to use ONNX for mmseg/mmdet models.